글 읽기 - [python2] 예외를 못찾겠습니다....자꾸 틀렸다고 나오네요..

lainroses 6년 전 0

도움 주시면 감사하겠습니다.

def dfs(graph, root):
    stack = [root]
    visited = [root]
    while stack:
        next_vertex = sorted(set(graph[stack[-1]]) - set(visited))
        if next_vertex:
            vertex = next_vertex[0]
            stack.append(vertex)
            visited.append(vertex)
        else:
            stack.pop()

    return visited

def bfs(graph, root):
    queue = [root]
    visited = [root]
    while queue:
        next_vertex = list(set(graph[queue.pop(0)]) - set(visited))
        if next_vertex:
            queue.extend(next_vertex)
            visited.extend(next_vertex)

    return visited

def get_graph(m):
    graph = {}
    for _ in range(m):
        a,b = map(int, raw_input().split())
        if not a in graph:
            graph[a] = [b]
        else:
            graph[a].append(b)
        if not b in graph:
            graph[b] = [a]
        else:
            graph[b].append(a)

    return graph

n,m,v = map(int, raw_input().split())
graph = get_graph(m)
print ' '.join(map(str, dfs(graph,v)))
print ' '.join(map(str, bfs(graph,v)))

lainroses 6년 전 0

BFS에 정렬이 안되있었네요..감사합니다

댓글을 작성하려면 로그인해야 합니다.

문제

문제

출처

ICPC

[python2] 예외를 못찾겠습니다....자꾸 틀렸다고 나오네요..

lainroses 6년 전 0

lainroses 6년 전 0

Baekjoon Online Judge

채점 현황

문제

유저 대회

고등학교 대회

출처

대학교 대회

도움말