본문 바로가기

알고리즘

12865 평범한 배낭 (JAVA)

평범한 배낭

문제

이 문제는 아주 평범한 배낭에 관한 문제이다.

한 달 후면 국가의 부름을 받게 되는 준서는 여행을 가려고 한다. 세상과의 단절을 슬퍼하며 최대한 즐기기 위한 여행이기 때문에, 가지고 다닐 배낭 또한 최대한 가치 있게 싸려고 한다.

준서가 여행에 필요하다고 생각하는 N개의 물건이 있다. 각 물건은 무게 W와 가치 V를 가지는데, 해당 물건을 배낭에 넣어서 가면 준서가 V만큼 즐길 수 있다. 아직 행군을 해본 적이 없는 준서는 최대 K만큼의 무게만을 넣을 수 있는 배낭만 들고 다닐 수 있다. 준서가 최대한 즐거운 여행을 하기 위해 배낭에 넣을 수 있는 물건들의 가치의 최댓값을 알려주자.

입력

첫 줄에 물품의 수 N(1 ≤ N ≤ 100)과 준서가 버틸 수 있는 무게 K(1 ≤ K ≤ 100,000)가 주어진다. 두 번째 줄부터 N개의 줄에 거쳐 각 물건의 무게 W(1 ≤ W ≤ 100,000)와 해당 물건의 가치 V(0 ≤ V ≤ 1,000)가 주어진다.

입력으로 주어지는 모든 수는 정수이다.

출력

한 줄에 배낭에 넣을 수 있는 물건들의 가치합의 최댓값을 출력한다.

풀이

다이나믹 알고리즘의 전형적인 문제중 하나인 배낭문제이다.
배낭 문제는 외워두는것이 좋다.
2차원 dp배열을 다음과 같이 선언한다.

dp[물건의 수][무게]

선언한 dp 배열을 채워주면 된다. dp 배열에 대해 조금 더 자세히 설명하자면 dp[1][5]는 0번물건과 1번물건을 가지고 무게 5까지의 최대 가치값이다.
점화식은 다음과 같다.
if(j >= 무게[i])
dp[i][j] = max(dp[i-1][j - 무게[i]] + 가치[i], dp[i-1][j])
// 해당 물건의 무게를 가진 후 현재 들 수 있는 무게j에서 무게[i]를 뺐을 때 구할수 있는 값의 합 vs 해당 물건을 가지지 않았을 경우
else if(j < 무게[i])
dp[i][j] = dp[i-1][j]

코드

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

public class Main{

    public static void main(String[] args) throws IOException {
        BufferedReader br = new BufferedReader(new InputStreamReader(System.in));
        StringTokenizer st = new StringTokenizer(br.readLine());

        int n = Integer.parseInt(st.nextToken());
        int m = Integer.parseInt(st.nextToken());

        int[][] dp = new int[n][m + 1];

        int[] weight = new int[n];
        int[] price = new int[n];

        for (int i = 0; i < n; i++) {
            st = new StringTokenizer(br.readLine());
            weight[i] = Integer.parseInt(st.nextToken());
            price[i] = Integer.parseInt(st.nextToken());
        }

        for (int i = 0; i <= m; i++) {
            if (i < weight[0]) {
                dp[0][i] = 0;
            }else{
                dp[0][i] = price[0];
            }
        }

        for (int i = 1; i < n; i++) {
            for (int j = 0; j <= m; j++) {
                if (weight[i] <= j) {
                    dp[i][j] = Math.max(price[i] + dp[i-1][j - weight[i]], dp[i-1][j]);
                }else{
                    dp[i][j] = dp[i-1][j];
                }
            }
        }


        System.out.println(dp[n-1][m]);
    }
}