[백준] 2606번(python 파이썬)

---

DFS, BFS 중 하나를 선택하여 해결할 수 있다.

 

BFS방식으로 문제를 해결하였다.

BFS는 큐를 이용하여 문제를 풀 수 있다.

1. 탐색 시작 노드를 큐에 삽입하고 방문 처리를 한다.

2. 큐에서 노드를 꺼내 해당 노드의 인접 노드 중 방문하지 않은 노드를 모두 큐에 삽입하고 방문처리를 한다.

3. 2번의 과정을 더 이상 수행할 수 없을 때까지 반복한다.

 

---

from collections import deque

n = int(input()); #컴퓨터 수
graph = [[] for _ in range(n + 1)]

e = int(input()); #연결된 선 수
for i in range(e):
    a, b = map(int, input().split())
    graph[a].append(b)
    graph[b].append(a)
    graph[a].sort()
    graph[b].sort()

visited = [False] * (n + 1)

def bfs(graph, start, visited):
    result = 0
    queue = deque([start])
    visited[start] = True

    while queue:
        pop = queue.popleft()
        for e in graph[pop]:
            if not visited[e]:
                queue.append(e)
                visited[e] = True
                result += 1
    print(result)

bfs(graph, 1, visited)