1325번 - 효율적인 해킹

https://www.acmicpc.net/problem/1325

1325번: 효율적인 해킹

#include <iostream>
#include <vector>
#include <cstring>
#include <algorithm>
using namespace std;
int N, M;
vector<int> relationship[10001];
bool isVisited[10001];
int ans = 1; //해킹할 수 있는 컴퓨터의 개수
int maxComputer = -1; //한 번에 몇 개의 컴퓨터를 해킹했는가
vector<int> number; //한 번에 가장 많은 컴퓨터를 해킹한 컴퓨터의 번호 저장
void dfs(int node)
{
    isVisited[node] = true;
    //신뢰관계 내에서
    for(int i=0; i<relationship[node].size(); i++)
    {
        int nextNode = relationship[node][i];
        if(isVisited[nextNode] == false)
        {
            isVisited[nextNode] = true;
            ans++;
            dfs(nextNode);
        }
    }
}
int main()
{
    cin >> N >> M;
    for(int i=0; i<M; i++)
    {
        int A, B;
        cin >> A >> B;
        relationship[B].push_back(A); //B가 해킹당하면 A도 해킹
    }
    for(int i=1; i<=N; i++)
    {
        //초기화
        memset(isVisited, false, sizeof(isVisited));
        ans = 1;

        dfs(i);

        if(ans == maxComputer)
        {
            number.push_back(i);
        }
        else if(ans > maxComputer)
        {
            maxComputer = ans; //update
            number.clear();
            number.push_back(i);
        }
    }

    sort(number.begin(), number.end());

    for(int i=0; i<number.size(); i++)
    {
        cout << number[i] << " ";
    }

    return 0;
}