코딩테스트/다이나믹 프로그래밍(Dynamic Programming))

[Java] 백준 1535번 : 안녕

sujin7837 2022. 4. 4. 22:31
반응형

문제

세준이는 성형수술을 한 후에 병원에 너무 오래 입원해 있었다. 이제 세준이가 병원에 입원한 동안 자기를 생각해준 사람들에게 감사하다고 말할 차례이다.

세준이를 생각해준 사람은 총 N명이 있다. 사람의 번호는 1번부터 N번까지 있다. 세준이가 i번 사람에게 인사를 하면 L[i]만큼의 체력을 잃고, J[i]만큼의 기쁨을 얻는다. 세준이는 각각의 사람에게 최대 1번만 말할 수 있다.

세준이의 목표는 주어진 체력내에서 최대한의 기쁨을 느끼는 것이다. 세준이의 체력은 100이고, 기쁨은 0이다. 만약 세준이의 체력이 0이나 음수가 되면, 죽어서 아무런 기쁨을 못 느낀 것이 된다. 세준이가 얻을 수 있는 최대 기쁨을 출력하는 프로그램을 작성하시오.

입력

첫째 줄에 사람의 수 N(≤ 20)이 들어온다. 둘째 줄에는 각각의 사람에게 인사를 할 때, 잃는 체력이 1번 사람부터 순서대로 들어오고, 셋째 줄에는 각각의 사람에게 인사를 할 때, 얻는 기쁨이 1번 사람부터 순서대로 들어온다. 체력과 기쁨은 100보다 작거나 같은 자연수 또는 0이다.

출력

첫째 줄에 세준이가 얻을 수 있는 최대 기쁨을 출력한다.

예제 입력 1

3
1 21 79
20 30 25

예제 출력 1

50

예제 입력 2

1
100
20

예제 출력 2

0

예제 입력 3

8
100 15 1 2 3 4 6 5
49 40 1 2 3 4 5 4

예제 출력 3

59

예제 입력 4

4
100 50 20 13
20 30 40 50

예제 출력 4

120

예제 입력 5

8
100 26 13 17 24 33 100 99
34 56 21 1 24 34 100 99

예제 출력 5

135

예제 입력 6

12
1 1 1 1 1 1 1 1 1 1 1 1
100 100 100 100 100 100 100 100 100 100 100 100

예제 출력 6

1200

 

소스코드

import java.io.BufferedReader;
import java.io.IOException;
import java.io.InputStreamReader;
import java.util.StringTokenizer;

public class Main {

	private static BufferedReader bf;
	private static StringTokenizer st;
	
	private static int N;
	private static int []L, J;
	private static int [][]dp;
	
	public static void main(String[] args) throws NumberFormatException, IOException {
		bf=new BufferedReader(new InputStreamReader(System.in));
		N=Integer.parseInt(bf.readLine());
		L=new int[N+1];
		J=new int[N+1];
		dp=new int[101][101];
		
		st=new StringTokenizer(bf.readLine());
		for(int i=1;i<=N;i++) L[i]=Integer.parseInt(st.nextToken());
		st=new StringTokenizer(bf.readLine());
		for(int i=1;i<=N;i++) J[i]=Integer.parseInt(st.nextToken());
		
		for(int i=1;i<=N;i++) {	// i번째 사람까지 고려했을 때
			for(int j=1;j<100;j++) {	// 체력이 j만큼 들 때 
				dp[i][j]=dp[i-1][j];	// 이전 값 저장
				if(j>=L[i]) dp[i][j]=Math.max(dp[i-1][j], dp[i-1][j-L[i]]+J[i]);	// 이전까지의 최댓값과 새로운 값을 고려했을 때의 값 중 더 큰 값으로 갱신
			}
		}
		System.out.println(dp[N][99]);
	}

}
사람 | 체력 1 2 3 4 ... 21 ... 99
1번째 20 20 20 20   20   20
2번째 20 20 20 20   30   30
3번째 30 30 30 30   30   50

체력을 100을 다 쓰면 죽기 때문에, 마지막 사람까지 고려하고 체력이 99까지 사용되었을 때인 dp[N][99] 값이 정답이 됩니다.

반응형