본문 바로가기
Algorithm/BFS, DFS

[ 백준] 2606번 바이러스

by 등촌동 꼬북이 2020. 9. 9.

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

 

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))

'Algorithm > BFS, DFS' 카테고리의 다른 글

[백준] 4963번 섬의 개수  (0) 2020.09.14
[백준] 7576번 토마토  (0) 2020.09.14
[백준] 1012번 유기농 배추  (0) 2020.09.09
[백준] 2644번 촌수계산  (0) 2020.09.08
[백준] 1926번 그림  (0) 2020.09.07

댓글