[백준] 파이썬 2606번 바이러스

- 문제: dfs/bfs 그래프 

https://www.acmicpc.net/problem/2606

2606번: 바이러스

- 코드:

정답코드를 보면서 dfs, bfs 방식이 어떻게 적용되는지 공부했다. dfs는 재귀함수(깊이), bfs는 큐(넓이)를 사용해서 진행된다. 

# dfs

n = int(input())
net = int(input())

graph = [[]*n for _ in range(n+1)]
for _ in range(net):
    a,b = map(int,input().split())
    graph[a].append(b)
    graph[b].append(a)

cnt = 0
visited = [0]*(n+1)

def dfs(start):
    global cnt
    visited[start] = 1
    for i in graph[start]:
        if visited[i]==0:
            dfs(i)
            cnt +=1
 
dfs(1)
print(cnt)

# bfs

n = int(input())
net = int(input())

graph = [[]*n for _ in range(n+1)]
for _ in range(net):
    a,b = map(int,input().split())
    graph[a].append(b)
    graph[b].append(a)

cnt = 0
visited = [0]*(n+1)

def bfs(start):
    global cnt
    visited[start] = 1
    queue = [start]
    print(queue)
    while queue:
        for i in graph[queue.pop()]:
            print(queue)
            if visited[i]==0:
                visited[i]=1
                print("visited: %s"%visited)
                queue.insert(0,i)  # queue 배열 맨앞에 놓기
                cnt +=1
 
bfs(1)
print(cnt)