[백준] 24480번 알고리즘 수업 - 깊이 우선 탐색 2 C++

문제 출처: https://www.acmicpc.net/problem/24480

24480번: 알고리즘 수업 - 깊이 우선 탐색 2

이전 문제와 동일하지만 내림차순으로 방문해야 하기 때문에 cmp함수를 정의해서 내림차순으로 정렬한 뒤 DFS 탐색 한다.

#include<iostream>
#include<algorithm>
#include<vector>
#include<queue>
#include<cstring>
#include <string>
#include <math.h>

using namespace std;


int N, M, R;
vector <int> adj[100001];
int visited[100001];
int Count = 1;

void dfs(int node) {
    
    for (auto next : adj[node]) {
        if (visited[next] != 0)continue;
        visited[next] = Count++;
        dfs(next);
    }
}

bool cmp(int a, int b) {
    return a > b;
}


int main() {
    ios::sync_with_stdio(false); cin.tie(NULL); cout.tie(NULL);

    cin >> N >> M >> R;
    for (int i = 0; i < M; i++) {
        int a, b;
        cin >> a >> b;
        adj[a].push_back(b);
        adj[b].push_back(a);
    }

    for (int i = 0; i <= N; i++) {
        sort(adj[i].begin(), adj[i].end(),cmp);
    }
    
    visited[R] = Count++;
    dfs(R);


    for (int i = 1; i <= N; i++) {
        cout << visited[i] << "\n";
    }

    

}