글 읽기 - 코드 어느 부분에서 잘못 했는지 모르겠습니다

comojin1994 4년 전 0

2579번 계단오르기 문제풀이를 기반으로 코드를 작성하였습니다.

https://www.acmicpc.net/problem/2579

코드 지적 해주시면 감사하겠습니다.

import sys
input = sys.stdin.readline

def func(N):
    if N == 1:
        print(val[0])
        return 1
    elif N == 2:
        print(val[0] + val[1])
        return 1
    elif N == 3:
        print(max(val[0] + val[2], val[1] + val[2]))
        return 1

    result = [val[0], val[0] + val[1], max(val[0] + val[2], val[1] + val[2])]

    for i in range(3, N):
        ##### 해당 잔에 도달하기 위해 바로 전 잔을 마시는 경우와 바로 전전 잔(b)을 마시는 경우
        ##### 바로 전 잔을 마실때는 i-3번째 잔까지 마신 것중 최대값(a) 선택
        a = max(result[:i-2])
        b = result[i-2]
        result.append(max(a + val[i-1] + val[i], b + val[i]))
    print(max(result[-1], result[-2]))

    return 1

if __name__ == '__main__':
    N = int(input())
    val = [int(input()) for _ in range(N)]
    func(N)