[BOJ/Python] 백준 2156번 - 포도주 시식

백준 #2156 포도주 시식

2156번: 포도주 시식

import sys
input = sys.stdin.readline

n = int(input())   # 포도잔 개수
wine = list(int(input()) for _ in range(n))

max_wine = [0 for _ in range(n)]

for i in range(n):
  if i == 0:
    max_wine[i] = wine[i]
  elif i == 1:
    max_wine[i] = max_wine[i-1] + wine[i]
  elif i == 2:
    max_wine[i] = max(max_wine[i-1], wine[i] + wine[i-2], wine[i] + wine[i-1])
  else:
    max_wine[i] = max(max_wine[i-1], wine[i] + max_wine[i-2], wine[i] + wine[i-1] + max_wine[i-3])

print(max_wine[-1])

 

max_wine 리스트에는 해당 포도주 위치까지 왔을 때 마실 수 있는 최대 포도주 양이 저장된다.

• 포도주를 안 마실 경우 : 이전 위치에서의 최대 포도주 양과 같다.
• 포도주를 마실 경우
  • 이전 포도주를 마실 경우 : (세 잔 연달아 마실 수 없기 때문에) 현재 포도주 양 + 두 잔 전 위치에서의 최대 포도주 양
  • 이전 포도주를 마시지 않을 경우 : 현재 포도주 양 + 이전 포도주 양 + 세 잔 전 위치에서의 최대 포도주 양