글 읽기 - 파이썬 출력 초과?...

ine 1년 전 0

메모제이션으로 풀었습니다.

if dp[idx] != -1:

에서 현재 위치 idx를 방문하기 위해 이전에 방문했던 위치를 visited 배열에 표시했습니다.

이 부분에서 조금 효율적이지 못하다는 생각이 들 수도 있는데요.

그것을 다 미뤄두고도 어디서 출력초과가 나는지 도저히 감을 잡을 수가 없네요.

정답을 출력하는 부분은 마지막

for i in dp: print(i)

인데, dp 배열 원소 개수만큼 출력하니까 초과해서 출력할 것이 보이질 않습니다 ㅠㅠ

코드에 존재하는 오류 지적해주시면 감사하겠습니다.

n = int(input())
a = [list(map(int,input().split())) for _ in range(n)]
dp = [-1]*n
def go(idx):
    global visited
    if dp[idx] != -1:
        for k in a[idx][1:]:
            if k == -1:
                break
            visited[k-1] = True
        return dp[idx]
    order = a[idx]
    time = order[0]
    for k in order[1:]:
        if k == -1:
            break
        if visited[k-1] == False:
            time += go(k-1)
    dp[idx] = time
    return dp[idx]
for i in range(n):
    visited = [False]*n
    go(i)
for i in dp:
    print(i)

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

문제

문제

출처

ICPC

파이썬 출력 초과?...

ine 1년 전 0

Baekjoon Online Judge

채점 현황

문제

유저 대회

고등학교 대회

출처

대학교 대회

도움말