Study/BOJ

BOJ 2156번

루ㅌ 2020. 2. 20. 18:00

DP 문제입니다.

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

 

2156번: 포도주 시식

효주는 포도주 시식회에 갔다. 그 곳에 갔더니, 테이블 위에 다양한 포도주가 들어있는 포도주 잔이 일렬로 놓여 있었다. 효주는 포도주 시식을 하려고 하는데, 여기에는 다음과 같은 두 가지 규칙이 있다. 포도주 잔을 선택하면 그 잔에 들어있는 포도주는 모두 마셔야 하고, 마신 후에는 원래 위치에 다시 놓아야 한다. 연속으로 놓여 있는 3잔을 모두 마실 수는 없다. 효주는 될 수 있는 대로 많은 양의 포도주를 맛보기 위해서 어떤 포도주 잔을 선택해야 할지 고

www.acmicpc.net

점화식을 잘 만드는게 중요하였습니다.

dp1, dp2, dp3이라는 변수를 만들어 놓고

for문을 돌면서 podojoo_list에서 다음 값을 계속해서 가져와서 max()를 가지고 가장 큰 값을 dp4에 담고 

dp4 -> dp3

dp3 -> dp2

dp2 -> dp1

와 같이 dp1,2,3을 업데이트 하였습니다. 

 


def maxdrink(podojoo_list, T):
	if T == 1:
		return podojoo_list[0]
	elif T == 2:
		return podojoo_list[0] + podojoo_list[1]
	else:
		dp1 = 0
		dp2 = podojoo_list[0]
		dp3 = podojoo_list[0] + podojoo_list[1]

		for index in range(2, T):
			dp4 = max([dp3, dp2 + podojoo_list[index], dp1 + podojoo_list[index-1] + podojoo_list[index]])
			
			dp1 = dp2
			dp2 = dp3
			dp3 = dp4
		return dp3

podojoo_list = []
import sys

T = int(sys.stdin.readline())
podojoo_list = [int(sys.stdin.readline()) for _ in range(T)]

print(maxdrink(podojoo_list, T))