글 읽기 - 무엇이 문제일까요(파이썬)

wbjkwjj 3년 전 0

반례도 다 통과하던데..ㅠㅠ

지적 부탁합니다

from sys import stdin
n=int(stdin.readline())
wine=[0]
for i in range(n):
    q=int(stdin.readline())
    wine.append(q)
if(n==1):
    result=wine[1]
elif(n==2):
    result=wine[1]+wine[2]
elif(n==3):
    result=max(max(wine[1]+wine[3],wine[2]+wine[3]),wine[1]+wine[2])
else:
    res=[[0,0],[wine[1],0],[wine[1]+wine[2],wine[1]],[max(wine[1]+wine[3],wine[2]+wine[3]),wine[1]+wine[2]]]#[if drink current one,or not]
    for i in range(4,n+1):
        d1=res[i-3][0]+wine[i-2]+wine[i]
        d2=res[i-3][0]+wine[i-1]+wine[i]
        d3=res[i-3][0]+wine[i]
        d4=res[i-3][1]+wine[i-2]+wine[i]
        d5=res[i-3][1]+wine[i-1]+wine[i]
        n1=res[i-3][0]+wine[i-2]
        n2=res[i-3][0]+wine[i-1]
        n3=res[i-3][1]+wine[i-2]+wine[i-1]
        res.append([max(d1,d2,d3,d4,d5),max(n1,n2,n3)])
    result=max(res[n][0],res[n][1])
print(result)

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

문제

문제

출처

ICPC

무엇이 문제일까요(파이썬)

wbjkwjj 3년 전 0

Baekjoon Online Judge

채점 현황

문제

유저 대회

고등학교 대회

출처

대학교 대회

도움말