[백준 16194]카드 구매하기2

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

16194번: 카드 구매하기 2

import sys

input = sys.stdin.readline

n = int(input())
a=[0]
arr = list(map(int,input().split()))
arr=a+arr
dp=arr

for i,v in enumerate(arr):
    if i != 0:
        for j in range(len(dp)-1, i-1, -1):
            dp[j]=min(dp[j], dp[j-i]+v)
print(dp[-1])

Python

 

"카드 구매하기 1" 코드 활용.

dp에 저장할 때 max함수 대신 min 함수를 써서 최저 값을 저장 함.