글 읽기 - 시간 초과가 뜨는데... 어디에서 뜨는질 모르겠습니다..

david02324 3년 전 0

나름 일관성있게 작성한다고 열심히 작성했고 테스트 케이스도 모두 통과했는데 시간 초과가 걸립니다. 어떤 반복문에서 무한 루프에 들어가는건가요..

import sys
from collections import deque
input = sys.stdin.readline

n,m,root = map(int,input().split())
nodes = [[] for _ in range(n+1)]
for _ in range(m):
    i,j = map(int,input().split())
    nodes[i].append(j)
    nodes[j].append(i)

for i in range(len(nodes)):
    nodes[i].sort()

def dfs(start,n):
    visited = [False] * (n+1)
    stack = []
    answerli = []

    visited[start] = True
    answerli.append(start)
    stack.append(start)
    while stack:
        for idx,node in enumerate(nodes[stack[-1]]):
            if not visited[node]:
                # 방문 처리
                visited[node] = True
                answerli.append(node)
                stack.append(node)
                break
            # 주변 노드가 모두 방문되었다면
            if idx == len(nodes[stack[-1]]) - 1:
                stack.pop()
    
    return answerli

def bfs(start,n):
    visited = [False] * (n+1)
    queue = deque()
    answerli = []

    visited[start] = True
    answerli.append(start)
    queue.append(start)
    while queue:
        for node in nodes[queue.popleft()]:
            if not visited[node]:
                # 방문 처리
                visited[node] = True
                answerli.append(node)
                queue.append(node)

    return answerli



print(' '.join(map(str,dfs(root,n))))
print(' '.join(map(str,bfs(root,n))))

shg9411 3년 전 2

다음과 같은 입력에서 무한루프가 발생합니다.

input
3 1 1
2 3

david02324 3년 전 0

감사합니다.. 선생님! 덕분에 깨달음을 얻었습니다!

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

문제

문제

출처

ICPC

시간 초과가 뜨는데... 어디에서 뜨는질 모르겠습니다..

david02324 3년 전 0

shg9411 3년 전 2

david02324 3년 전 0

Baekjoon Online Judge

채점 현황

문제

유저 대회

고등학교 대회

출처

대학교 대회

도움말