글 읽기 - 정렬까지 해줬는데 어디가 문제인지.. 감이 안잡힙니다ㅠ

725psh 4년 전 0

예제들이랑 제가 해본 테케들은 잘 나오는데...

어는 부분에서 꼬인건지 계속 틀렸다고 나오네요ㅠㅠ

감이 잘 안잡힙니다..ㅠ

class Graph:
    class Node:
        data = None
        adj = None
        mark = None
        def __init__(self, _data):
            self.data = _data
            self.adj = list()
            self.mark = False

    nodes = []

    def __init__(self, _N:int):
        for n in range(_N):
            self.nodes.append(self.Node(n+1))


    def addEdge(self, _a:int, _b:int):
        # 인덱스 맞추기
        n1 = self.nodes[_a - 1]
        n2 = self.nodes[_b - 1]

        if n2 not in n1.adj:
            n1.adj.append(n2)
        if n1 not in n2.adj:
            n2.adj.append(n1)


    def setMarkFalse(self):
        for i in range(len(self.nodes)):
            self.nodes[i].mark = False
        print()


    def visit(self, _temp):
        print(_temp.data, end=" ")


    def dfs(self, _V):
        root = self.nodes[_V - 1] #인덱싱
        stack = [ root ]
        root.mark = True

        while len(stack) != 0:
            temp = stack.pop()
            for a in temp.adj:
                if a.mark is False:
                    a.mark = True
                    stack.append(a)

            self.visit(temp)


    def bfs(self, _V):
        root = self.nodes[_V - 1]
        queue = [ root ]
        root.mark = True

        while len(queue) != 0:
            temp = queue.pop(0)
            for a in temp.adj:
                if a.mark is False:
                    a.mark = True
                    queue.append(a)

            self.visit(temp)


def main():
    N, M, V = map(int, input().split())
    g = Graph(N)

    edges = []
    for m in range(M):
        edges.append(list(map(int, input().split())))

    edges.sort(key= lambda e : e[0])
    edges.sort(key= lambda e : e[1])

    for e in edges:
        g.addEdge(e[0], e[1])


    g.dfs(V)
    g.setMarkFalse()
    g.bfs(V)


if __name__ == '__main__':
    main()

bupjae 4년 전 0

입력이 다음과 같을 때

정답:

1 2 4 3
1 2 3 4

오답:

1 4 3 2
1 2 3 4

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

문제

문제

출처

ICPC

정렬까지 해줬는데 어디가 문제인지.. 감이 안잡힙니다ㅠ

725psh 4년 전 0

bupjae 4년 전 0

Baekjoon Online Judge

채점 현황

문제

유저 대회

고등학교 대회

출처

대학교 대회

도움말