210616 - TIL

김예지·2021년 6월 16일
0

dfs와 bfs를 계속 적용해보려고 노력중이다.
소스코드를 외우는게 더 빠를것같다는 생각도 든다.

def dfs(v):
    print(v, end=' ')
    visit[v] = 1
    for i in range(1, n + 1):
        if visit[i] == 0 and s[v][i] == 1:
            dfs(i)

def bfs(v):
    queue = [v]
    visit[v] = 0
    while(queue):
        v = queue[0]
        print(v, end=' ')
        del queue[0]
        for i in range(1, n + 1):
            if visit[i] == 1 and s[v][i] == 1:
                queue.append(i)
                visit[i] = 0

n, m, v = map(int, input().split())
s = [[0] * (n + 1) for i in range(n + 1)]
visit = [0 for i in range(n + 1)]
for i in range(m):
    x, y = map(int, input().split())
    s[x][y] = 1
    s[y][x] = 1
    
dfs(v)
print()
bfs(v)

dfs는 깊이 우선 탐색
bfs는 너비 우선 탐색
소스코드를 이해못하겠으면 외워버리자.

profile
새싹

0개의 댓글