210615 - TIL

백준 1260번 - 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를 한 결과를 출력하면 된다.
자료구조 시간에 개념만 익혔더니 코드 구현이 힘들어 구글신의 도움을 받았다.
dfs는 깊이우선탐색으로, 다음꺼 방문 안했으면 걍 계속 방분한다.
bfs는 너비우선탐색으로, 옆에꺼 방문 안했으면 큐에 넣는다... 그 이상은 아직 이해중이다.