[ 파이썬(python) ] 백준 2606 - 바이러스(DFS)

📌 백준 2606 - 바이러스(DFS)

문제 설명

---

💡 나의 풀이

코드 업에서 풀었던 문제인데, 다시 풀어봤다. DFS를 이용해서 방문하지 않은 노드가 있을 때 cnt를 누적하는 방법으로 구현했다.

중요한 점은 cnt를 전역변수로 사용했기 때문에 파라미터로 전달하지 않았다.

728x90

import sys
input = sys.stdin.readline

def dfs(graph, v, visited):
    global cnt
    visited[v] = True
    for i in graph[v]:
        if not visited[i]:
            cnt += 1
            dfs(graph, i, visited)

PC = int(input())
network = int(input())

graph = [[] for _ in range(PC + 1)]
visited = [False] * (PC + 1)
cnt = 0    # global variable

for i in range(network):
    node1, node2 = map(int, input().split())
    graph[node1].append(node2)
    graph[node2].append(node1)

dfs(graph, 1, visited)
print(cnt)