글 읽기 - 시간초과가 뜨는데 어떻게 해결해야 할까요?

k550706 1년 전 1

55% 정도에서 시간초과가 뜹니다

나름대로 잘 작성한 같은데, 어디서 잘못된건지 모르겠습니다 ㅠㅠ

고수님들 조언 부탁드립니다.

import sys
sys.setrecursionlimit(150000)
n,r,q=map(int,sys.stdin.readline().rstrip().split())

graph=[[] for i in range(n+1)]
visited=[0 for i in range(n+1)]
for i in range(n-1):
    a,b=map(int,sys.stdin.readline().rstrip().split())
    graph[a].append(b)
    graph[b].append(a)

def DFS(v):
    visited[v]+=1
    for i in graph[v]:
        if visited[i]==0:
            visited[v]+=DFS(i)
    return visited[v]

DFS(r)

for i in range(q):
    a=int(input())
    print(visited[a])

k550706 1년 전 1

자문자답입니다

마지막 print에서 input 대신 sys.stdin.readline()으로 처리하면 맞습니다.

flip1945 1년 전 0

덕분에 쉽게 풀었습니다.

맞왜틀 계속하다가 input()이 오래걸리는 걸 까먹었네요

감사합니다~!

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

문제

문제

출처

ICPC

시간초과가 뜨는데 어떻게 해결해야 할까요?

k550706 1년 전 1

k550706 1년 전 1

flip1945 1년 전 0

Baekjoon Online Judge

채점 현황

문제

유저 대회

고등학교 대회

출처

대학교 대회

도움말