백준 2458번 키 순서 C++

출처 : https://www.acmicpc.net/problem/2458

 

자기보다 작은 번호를 연결하는 그래프와 큰 번호를 연결하는 그래프 두개를 만들어서 둘다 순회했을 때 모든 노드를 방문할 수 있다면 자신의 키가 몇 번째인지 알 수 있다.

 

#include <iostream>
#include <string>
#include <vector>
#include <math.h>
#include <algorithm>
#include <utility>
#include <stack>
#include <queue>
#include <math.h>
#include <set>
#include <map>
#include <list>
#include <unordered_map>
#include <unordered_set>
#include <iomanip>
#include <limits.h>

#define OUT

using namespace std;
using int64 = long long;

int n, m;

vector<vector<int>> graph1;
vector<vector<int>> graph2;
vector<bool> visited;

void Dfs(int n, set<int>& s, vector<vector<int>>& graph) {
    visited[n] = true;
    s.insert(n);
    for (auto next : graph[n]) {
        if (visited[next]) continue;
        Dfs(next, s, graph);
    }
}

void RefreshVisited() {
    for (int i = 1; i <= n; i++) {
        visited[i] = false;
    }
}

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

    cin >> n >> m;
    graph1 = vector<vector<int>>(n + 1, vector<int>());
    graph2 = vector<vector<int>>(n + 1, vector<int>());
    visited = vector<bool>(n + 1, false);
    for (int i = 0; i < m; i++) {
        int a, b;
        cin >> a >> b;
        graph1[a].push_back(b);
        graph2[b].push_back(a);
    }
    int ans = 0;
    for (int i = 1; i <= n; i++) {
        set<int> s;
        RefreshVisited();
        Dfs(i, s, graph1);
        RefreshVisited();
        Dfs(i, s, graph2);
        if (s.size() == n) ans++;
    }
    cout << ans;
}