글 읽기 - KeyError 질문

hshh3 2년 전 0

vscode에서는 입출력이 잘 되는데

백준에서 제출을 하면 keyerror가 뜨네요...

해결부탁드립니다

class Graph():
    def __init__(self):
        self.graph = {}

    def addEdge(self, startV, endV):
        if startV not in self.graph:
            self.graph[startV] = []
        if endV not in self.graph:
            self.graph[endV] = []
        
        self.graph[startV].append(endV)
        self.graph[endV].append(startV)

    def bfs(self, startV):
        q = [startV]
        visited = [startV]
        while q:
            nowV = q.pop(0)
            for nextV in self.graph[nowV]:
                if nextV not in visited:
                    q.append(nextV)
                    visited.append(nextV)
        return visited

    def dfs(self, startV):
        s = [startV]
        visited = []
        while s:
            nowV = s.pop()
            if nowV not in visited:
                visited.append(nowV)
                s.extend(self.graph[nowV][::-1])
        return visited
    
    def dfs_recursive(self, startV, visited=[]):
        visited.append(startV)

        for nextV in self.graph[startV]:
            if nextV not in visited:
                self.dfs_recursive(nextV, visited)

        return visited

g = Graph()
import sys
n, m, v = map(int, input().split())
for _ in range(m):
    a, b = map(int, sys.stdin.readline().split())
    g.addEdge(a, b) 

for i in g.graph:
    g.graph[i].sort()

for i in g.dfs(v):
    print(i, end=' ')
print()
for i in g.bfs(v):
    print(i, end=' ')
print()

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

문제

문제

출처

ICPC

KeyError 질문

hshh3 2년 전 0

Baekjoon Online Judge

채점 현황

문제

유저 대회

고등학교 대회

출처

대학교 대회

도움말