백준 14501 [퇴사] 파이썬

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

14501번: 퇴사

n = int(input())
T = [0]
P = [0]
dp = [0]*(n+2)
for _ in range(n):
    cost, pay = map(int, input().split())
    T.append(cost)
    P.append(pay)

for k in range(1, n+2):
    for j in range(1, k):
        if k == j+T[j]:
            dp[k] = max(dp[j]+P[j], dp[k], dp[k-1])
        else:
            dp[k] = max(dp[k-1], dp[k])
print(dp[n+1])

먼저 드는 시간과 얻는 비용을 각각 T와 P 리스트에 append한다.

그 후에, 이중 for문을 통해 각 날짜별로 최적의 해를 계산한다.

 

다른분들의 풀이를 보니, n일부터 내려가는 형식으로 푸신분도 계셨다.

#14501
n=int(input())
t=[]
p=[]
dp=[0]*(n+1)

for i in range(n):
    ts, ps= map(int,input().split())
    t.append(ts)
    p.append(ps)
    
for i in range(n-1,-1,-1):
    if i+t[i]>n:
        dp[i]=dp[i+1]
    else:
        dp[i]=max(dp[i+1],p[i]+dp[i+t[i]])
print(dp[0])