Programming/BackJoon
2156 - 포도주 시식
홍열
2023. 3. 14. 14:39
728x90
1) 문제 설명
- 포도주를 마시는데, 최대로 마실 수 있는 양을 구하는 문제
- 단, 조건은 3잔 연속 마실 수 없다.
- 예를 들어서, 1,2,3 의 포도주가 있으면
1+2+3 => 불가 (3잔 연속으로 마실 수 없음)
1+ 3 => 가능
2+3 => 가능
- 위 경우 답은 5이다.
문제에 나온 케이스를 가지고 분석해보면,
2) 문제 구현
- dp[0]은 입력받은 값의 0번째가 무조건 들어간다. (dp[0] = input[0])
- 1일 때는 max(input[1] + dp[0])
- 2일 때는 고려할게 하나 더 늘어난다. 3번 연속 마시는 경우를 고려해야한다.
input[2]+input[1], input[2] + input[0], dp[1]
2번째와 첫번째 잔을 마시거나
2번째와 0번째 잔을 마시거나
2번째를 마시지 않고 그 전만 마시거나
- 3일때는 고려할 사항 한 가지 더 늘어난다.
현재 잔을 마시지 않고 , 이전껄로 만족하거나 (d[i-1])
현재 잔을 마시고, 2번째 전껄로 더한다. (input[i] + dp[i-2])
현재잔 마시고, 이전잔 마시고, 그 이이전의 합을 취한다.(input[i] + input[i-1] + d[i-3])
이렇게 하면 건너뛰면서 마실 수 있다.
import sys
t = int(input())
lst = []
for _ in range(t):
lst.append(int(input()))
dp = [0] * (t)
dp[0] = lst[0]
for i in range(1, t):
if i == 1:
dp[i] = lst[0] + lst[1]
elif i == 2:
dp[i] = max(lst[2] + lst[1], lst[2] + lst[0], dp[1])
else:
dp[i] = max(dp[i-1], dp[i-3] + lst[i-1] + lst[i] , dp[i-2] + lst[i])
print(dp[t-1])