[파이썬/백준 1260]DFS와 BFS

문제

그래프를 DFS로 탐색한 결과와 BFS로 탐색한 결과를 출력하는 프로그램을 작성하시오. 단, 방문할 수 있는 정점이 여러 개인 경우에는 정점 번호가 작은 것을 먼저 방문하고, 더 이상 방문할 수 있는 점이 없는 경우 종료한다. 정점 번호는 1번부터 N번까지이다.

첫째 줄에 정점의 개수 N(1 ≤ N ≤ 1,000), 간선의 개수 M(1 ≤ M ≤ 10,000), 탐색을 시작할 정점의 번호 V가 주어진다. 다음 M개의 줄에는 간선이 연결하는 두 정점의 번호가 주어진다. 어떤 두 정점 사이에 여러 개의 간선이 있을 수 있다. 입력으로 주어지는 간선은 양방향이다.

첫째 줄에 DFS를 수행한 결과를, 그 다음 줄에는 BFS를 수행한 결과를 출력한다. V부터 방문된 점을 순서대로 출력하면 된다.

 

주의할 점

이건 진짜 지옥같은 문제다. 도저히 왜 틀렸는지 납득하지 못하다가 출력 포맷이 잘못되었다는 것을 깨달았다.

출력 포맷을 그냥 보면 모르겠지만, 드래그해보면 공백이 붙어있다. 공백을 붙여주면 정답이 된다. 설마 정말 진짜 이거때문에 틀린거였는지 확인하기위해 마지막에 공백을 더한 것을 제거해봤더니.. 틀렸다고 나왔다. 문제는 공백이었던 것이다.

 

그리고 문제에 방문할 수 있는 정점이 여러 개인 경우에는 정점 번호가 작은 것을 먼저 방문해야 한다고 나와있다.

예제를 처음 테스트할 때는 재귀 DFS와 반복 DFS의 차이인줄 알았는데, 정점 번호가 작은 것 먼저 방문하게 처리를 해야했다.

 

코드

# n 정점의 개수, m 간선의 개수, v 탐색을 시작할 정점의 번호
# m번 입력

def main():
    from collections import defaultdict

    n, m, v = map(int, input().split())

    graph = defaultdict(list)
    for _ in range(m):
        node, edge = map(int, input().split())
        graph[node].append(edge)
        graph[edge].append(node)

    for node in list(graph):
        graph[node] = sorted(graph[node])

    def recursive_dfs(v, discovered = []):
        discovered.append(v)
        for w in graph[v]:
            if w not in discovered:
                discovered = recursive_dfs(w, discovered)
        return discovered

    def bfs(v):
        queue, discovered = [v], [v]
        while queue:
            v = queue.pop(0)
            for w in graph[v]:
                if w not in discovered:
                    discovered.append(w)
                    queue.append(w)
        return discovered

    print(' '.join(map(str, recursive_dfs(v)))+' ')
    print(' '.join(map(str, bfs(v)))+' ')

if __name__ == '__main__':
    main()

 

 

링크

www.acmicpc.net/problem/1260

1260번: DFS와 BFS