Algorithm/BFS, DFS

[ 백준] 2606번 바이러스

등촌동 꼬북이 2020. 9. 9. 04:12

이 문제도 너무 쉬운 문제..

 

from collections import deque

def bfs(graph):
    visited = []
    que = deque([1])
    while que:
        t = que.popleft()
        if t in visited:
            continue
        if not t in visited:
            visited.append(t)
            que.extend(sorted(graph[t]))
    if len(visited) == 1:
        return 1
    else:
        return len(visited) - 1

N = int(input())
M = int(input())

graph = [set([]) for _ in range(N + 1)]

for _ in range(M):
    V1, V2 = map(int, input().split())
    graph[V1].add(V2)
    graph[V2].add(V1)

print(bfs(graph))