백준 1260 DFS와 BFS

born_a·2022년 11월 8일
0
post-thumbnail

def dfs(n):
    #시작노드 방문 처리
    visited[n] = True
    #시작노드 출력
    print(n, end=' ')
    #인접 노드 방문 하였는지 확인
    for i in graph[n]:
        if visited[i] == True:
            continue
        #방문하지 않았으면
        else:
            visited[i] = True
            dfs(i)

def bfs(n):
    queue = deque([n])
    visited[n] = True
    while queue:
        v = queue.popleft()
        print(v, end=' ')
        for i in graph[v]:
            if not visited[i]:
                queue.append(i)
                visited[i] = True


from collections import deque
n, m, v = (map(int, input().split()))
visited = [False] * (n+1)
graph = [[] for _ in range(n+1)]
for i in range(m):
    a,b = map(int,input().split())
    graph[a].append(b)
    graph[b].append(a)
for i in range(1,n+1):
    graph[i].sort()   
print(graph)

dfs(v)
print()

#방문 기록 초기화를 시켜줘야함...
visited = [False] * (n+1)
bfs(v)

dfs, bfs 함수 짜는건 수월했으나, 그래프에 노드들을 저장할 때의 아이디어가 부족했음.
마지막에 다 구현해두고 bfs(v)가 1이 출력되어서 에러를 찾고자 고생했는데, 알고보니 dfs(v)에서 이미 방문처리를 다 해두어서 1만 출력된것이었음.
visited리스트 요소들을 모두 False처리를 해주면서 해결함.

0개의 댓글