개발자가 되고 싶은 준개발자

[백준] 퇴사 본문

알고리즘 공부/삼성 역량테스트

[백준] 퇴사

준개발자 2020. 9. 20. 10:57

풀이

작은 문제의 계산 결과를 재사용하여 큰 문제의 결과를 구할 수 있으므로 동적 프로그래밍을 이용하여 풀 수 있다. 

특정 날짜에서 받을 수 있는 비용의 최댓값은 그 이전 날짜 중에서 받을 수 있는 비용의 최댓값에 현재 날짜의 비용을 더한 값이 될 수 있다. 단 주의해야 할 점은 '상담을 하는데 필요한 기간은 1일보다 클 수 있기 때문에, 모든 상담을 할 수는 없다.'라는 제약사항을 고려해야 한다. 따라서 특정 날짜에서 받을 수 있는 비용의 최댓값을 구하기 위해서는 이전 날짜 중에서 최댓값을 더하되, 이전 날짜의 상담이 특정 날짜에는 종료되어 있어야 한다.


코드

import sys 

# 입력 읽기
n = int(sys.stdin.readline())
t, p, dp = [0 for _ in range(n)], [0 for _ in range(n)], [0 for _ in range(n)]
for i in range(n):
    line = sys.stdin.readline()
    line = line.split()
    t[i], p[i] = int(line[0]), int(line[1])

#동적 프로그래밍으로 최댓값 계산
for i in range(n):
    if i + t[i] <= n:
        dp[i], curMax = p[i], p[i]
        for j in range(0, i): 
            if j+t[j] <= i:
                curMax = max(curMax, dp[i]+dp[j])
        dp[i] = curMax
        
print(max(dp))

백준 제출 결과


출처

www.acmicpc.net/problem/14501