-
2156 - 포도주 시식Programming/BackJoon 2023. 3. 14. 14:39728x90
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])
'Programming > BackJoon' 카테고리의 다른 글
1699 - 제곱수의 합 (0) 2023.03.17 2579 - 계단 오르기 (0) 2023.03.16 1912 - 연속합 (1) 2023.03.15 1966 - 프린터 큐 (1) 2023.03.15 9465 - 스티커 (0) 2023.03.14