백준 1325 효율적인 해킹 (파이썬)

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

1325번: 효율적인 해킹

1. 양방향이 아니라 단방향으로 관계를 만들어줍니다

2. bfs로 경로를 따라 이동하면서 거리를 저장하고 최대값에 해당하는 인덱스를 오름차순으로 출력

from collections import deque
import sys

input = sys.stdin.readline

def bfs(x):
    q = deque()
    c = [0 for _ in range(n)]
    q.append(x)
    c[x] = 1
    cnt = 1
    while q:
        x = q.popleft()
        for nx in a[x]:
            if c[nx] == 0:
                cnt += 1
                c[nx] = 1
                q.append(nx)
    return cnt

n, m = map(int, input().split())
a = [[] for _ in range(n)]

for _ in range(m):
    x, y = map(int, input().split())
    a[y-1].append(x-1)

ans = [0 for _ in range(n)]
for i in range(n):
    ans[i] = bfs(i)
for i in range(n):
    if ans[i] == max(ans):
        print(i+1, end=' ')