[백준] 14501 - 퇴사 (자바 Java)

[Silver III] 퇴사 - 14501

문제 링크

성능 요약

메모리: 12892 KB, 시간: 108 ms

분류

브루트포스 알고리즘(bruteforcing), 다이나믹 프로그래밍(dp)

문제 설명

상담원으로 일하고 있는 백준이는 퇴사를 하려고 한다.

오늘부터 N+1일째 되는 날 퇴사를 하기 위해서, 남은 N일 동안 최대한 많은 상담을 하려고 한다.

백준이는 비서에게 최대한 많은 상담을 잡으라고 부탁을 했고, 비서는 하루에 하나씩 서로 다른 사람의 상담을 잡아놓았다.

각각의 상담은 상담을 완료하는데 걸리는 기간 T_i와 상담을 했을 때 받을 수 있는 금액 P_i로 이루어져 있다.

N = 7인 경우에 다음과 같은 상담 일정표를 보자.

	1일	2일	3일	4일	5일	6일	7일
Ti	3	5	1	1	2	4	2
Pi	10	20	10	20	15	40	200

1일에 잡혀있는 상담은 총 3일이 걸리며, 상담했을 때 받을 수 있는 금액은 10이다. 5일에 잡혀있는 상담은 총 2일이 걸리며, 받을 수 있는 금액은 15이다.

상담을 하는데 필요한 기간은 1일보다 클 수 있기 때문에, 모든 상담을 할 수는 없다. 예를 들어서 1일에 상담을 하게 되면, 2일, 3일에 있는 상담은 할 수 없게 된다. 2일에 있는 상담을 하게 되면, 3, 4, 5, 6일에 잡혀있는 상담은 할 수 없다.

또한, N+1일째에는 회사에 없기 때문에, 6, 7일에 있는 상담을 할 수 없다.

퇴사 전에 할 수 있는 상담의 최대 이익은 1일, 4일, 5일에 있는 상담을 하는 것이며, 이때의 이익은 10+20+15=45이다.

상담을 적절히 했을 때, 백준이가 얻을 수 있는 최대 수익을 구하는 프로그램을 작성하시오.

입력

첫째 줄에 N (1 ≤ N ≤ 15)이 주어진다.

둘째 줄부터 N개의 줄에 T_i와 P_i가 공백으로 구분되어서 주어지며, 1일부터 N일까지 순서대로 주어진다. (1 ≤ T_i ≤ 5, 1 ≤ P_i ≤ 1,000)

출력

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

---

풀이

 

처음에는 idx의 완료일 이후의 index로 가서 해당 값과 dp값을 비교해서 update하면 된다고 생각했다.

그런데 문제 입력 4번을 보니 

10
5 50
4 40
3 30
2 20
1 10
1 10
2 20
3 30
4 40
5 50

마지막에 2일짜리 20을 더하는 것보다 3일짜리 30을 더 하는 게 더 큰 경우가 존재했다.

그래서 idx에서 다음 index(next)로 넘어가고, next에서 N-1까지 검토하여 추가 가능한 index가 있다면 모두 update를 해주었다.

 

package BOJ.Silver.s3;

import java.util.Arrays;
import java.util.Scanner;

public class BOJ_14501_퇴사 {
    public static void main(String[] args) {
        Scanner sc = new Scanner(System.in);
        int N = sc.nextInt();
        int[][] arr = new int[N][2];
        int[] dp = new int[N];
        for (int i = 0; i < N; i++) {
            int day = sc.nextInt();
            int profit = sc.nextInt();
            arr[i][0] = day;
            arr[i][1] = profit;
        }
        for (int i = 0; i < N; i++) {
            if (i + arr[i][0] > N) continue; // 완료일이 N을 넘으면 고려X
            dp[i] = Math.max(dp[i], arr[i][1]);
            int idx = i;
            while (idx < N) { // 다음날 검사
                int next = idx + arr[idx][0]; 
                for (int j = next; j < N; j++) // idx의 완료일 이후에 수행하는 작업
                    if (j + arr[j][0] <= N) { // next의 작업이 N일 내에 끝난다면
                        dp[j] = Math.max(dp[j], dp[idx] + arr[j][1]); // 답 update
                    }
                idx = next; // 다음 idx 검사
            }

        }
        int ans = 0;
        for (int i = 0; i < N; i++)
            ans = Math.max(dp[i], ans);
        System.out.println(ans);

    }
}