[백준] 2606 바이러스

import sys
input = sys.stdin.readline

n = int(input()) # 정점 수
m = int(input()) # 간선 정보 수
graph = [ [] for _ in range(n+1) ]
visited = [0] * (n+1)
cnt = 0

for _ in range(m):
    a, b = map(int, input().split())
    graph[a].append(b)
    graph[b].append(a)

for lst in graph:
    lst.sort()

def dfs(graph, r, visited):
    global cnt
    visited[r] = True # 방문 처리
    cnt+=1
    for i in graph[r]:
        if visited[i] == 0:
            dfs(graph, i, visited)

dfs(graph, 1, visited)
print(cnt-1) # 1번 컴퓨터는 제외하므로

다시 풀기

1
첫번째 탐색 노드(1)의 경우는 감염된 컴퓨터의 수에 포함시키지 않기 때문에 그 밑의 for 문에서부터 cnt 를 1씩 더해준다.

import sys
sys.setrecursionlimit(10**6)
input = sys.stdin.readline

n = int(input().strip())
m = int(input().strip())

visited = [0] * (n+1)
graph = [[] for _ in range(n+1)]

for _ in range(m):
    n1,n2 = map(int,input().split())
    graph[n1].append(n2)
    graph[n2].append(n1)

for arr in graph:
    arr.sort()

cnt = 0
def dfs_rec(graph, visited, r):
    visited[r] = 1

    for nv in graph[r]:
        if visited[nv] == 0:
            visited[nv] = 1
            global cnt
            cnt+=1
            dfs_rec(graph, visited, nv)     

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