[백준] S2 1535번 안녕 (JAVA)

문제 출처 - Baekjoon Online Judge

문제는 여기

1535번: 안녕

[문제] 

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

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

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

[입력]

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

[출력]

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

 

---

[풀이]

1. n을 입력 받아준다.

2. 인사할 때 잃는 체력을 입력받아준다.

3. 인사할 때 얻는 행복을 입력받아준다.

4. 인사할 때 드는 체력이 100이 안 넘어가는 선에서 각 hp를 빼고 행복을 더한 것과 현재의 기쁨을 비교해 더 큰 것으로 갱신해준다.

5. 100을 넘어가면 안 되므로 99일 때의 값을 출력해준다.

[접근]

1. Knapsack문제를 풀 듯 풀면 되겠다고 생각하였다.

[코드]

import java.io.*;
import java.util.*;

public class Main {
	public static void main(String[] args) throws IOException{
		BufferedReader br = new BufferedReader(new InputStreamReader(System.in));
		int n = Integer.parseInt(br.readLine());
		int[] dp = new int[100];
		int[] hp = new int[n];
		int[] happy = new int[n];
		
		// 인사할 때 잃는 체력 입력
		StringTokenizer st = new StringTokenizer(br.readLine());
		for (int i = 0; i < n; i++) {
			hp[i] = Integer.parseInt(st.nextToken());
		}
		
		// 인사할 때 얻는 행복 입력
		st = new StringTokenizer(br.readLine());
		for (int i = 0; i < n; i++) {
			happy[i] = Integer.parseInt(st.nextToken());
		}
		
		for (int i = 0; i < n; i++) {
			// 인사할 때 드는 체력이 100이 안넘어가는 선에서
			for (int j = 99; j >= hp[i]; j--) {
				// 각 hp를 빼고 행복을 더한 것 vs 현재까지 기쁨
				// 비교해서 더 큰 값을 고르기
				dp[j] = Math.max(dp[j- hp[i]] + happy[i], dp[j]);
			}
		}
		
		// 100이 되면 안되므로 99일 때 최대값을 출력 
		System.out.println(dp[99]);
	}
}