글 읽기 - 런타임에러가 자꾸 나는데 뭐가 문제인지 모르겠어요 ㅠㅠㅠ

andrewkimber 6년 전 0

우선... 혼자서 맨땅에 헤딩식으로 공부 중이라, 코드가 많이 지저분해도 너그러이 봐주세요 ㅠㅠ

제 논리는

dp(n)을 구할 때,

1. n번째 포도주를 마시지 않는 경우

2. n번째 포도주와 n-1번째 포도주를 마시는 경우

3. n번째 포도주는 마시고 n-1번째 포도주를 마시지 않는 경우

중 maximum 값을 구한다

입니다...

논리 상으로는 틀린 게 없는 것 같고, 몇가지 예제를 시도해봐도 제대로 작동하는 것 같은데 런타임에러가 나네요.... ㅠㅠ

N = int(raw_input())

wine = [0]
for i in range(N):
    wine_cost = int(raw_input())
    wine.append(wine_cost)


def dp(n, memo={}):
    if n in memo:
        return memo[n]
    elif n == 1:
        memo[1] = wine[1]
        return memo[1]
    elif n == 2:
        memo[2] = wine[1] + wine[2]
        return memo[2]
    elif n == 3:
        memo[3] = max(wine[2] + wine[3], wine[1] + wine[3], wine[1] + wine[2])
        return memo[3]
    else:
        temp = max(dp(n - 3, memo) + wine[n - 1] + wine[n], dp(n - 2, memo) + wine[n], dp(n - 1, memo))
        memo[n] = temp
        return temp

print dp(N)

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

문제

문제

출처

ICPC

런타임에러가 자꾸 나는데 뭐가 문제인지 모르겠어요 ㅠㅠㅠ

andrewkimber 6년 전 0

Baekjoon Online Judge

채점 현황

문제

유저 대회 / 고등학교 대회

출처

대학교 대회

도움말