백준(Python) 풀이/다이나믹 프로그래밍

백준(Python) 14501번 퇴사 풀이

개발윗미 2022. 1. 12. 15:24

Python으로 구현한 14501번 퇴사 문제 풀이입니다.

 

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

 

14501번: 퇴사

첫째 줄에 백준이가 얻을 수 있는 최대 이익을 출력한다.

www.acmicpc.net


n = int(input()) # 전체 상담 개수
t = [] # 각 상담을 완료하는 데 걸리는 기간
p = [] # 각 상담을 완료했을 때 받을 수 있는 금액
dp = [0] * (n + 1) # 다이나믹 프로그래밍을 위한 1차원 dp 테이블 초기화
max_value = 0

for _ in range(n) :
  x, y = map(int, input().split())
  t.append(x)
  p.append(y)

# 리스트를 뒤에서부터 거꾸로 확인
for i in range(n - 1, -1, -1) :
  time = t[i] + i
  # 상담이 기간 안에 끝나는 경우
  if time <= n :
    # 점화식에 맞게, 현재까지의 최고 이익 계산
    dp[i] = max(p[i] + dp[time], max_value)
    max_value = dp[i]
  
  # 상담이 기간을 벗어나는 경우
  else :
    dp[i] = max_value

print(max_value)