華爲機試題01揹包問題

package com.nowcoder.huawei;

import java.util.Scanner;

/**
 * 輸入的第 1 行，爲兩個正整數，用一個空格隔開：N m
 * （其中 N （ <32000 ）表示總錢數， m （ <60 ）爲但願購買物品的個數。）
 *
 *
 * 從第 2 行到第 m+1 行，第 j 行給出了編號爲 j-1 的物品的基本數據，每行有 3 個非負整數 v p q
 * （其中 v 表示該物品的價格（ v<10000 ），
 * p 表示該物品的重要度（ 1 ~ 5 ）， q 表示該物品是主件仍是附件。
 * 若是 q=0 ，表示該物品爲主件，若是 q>0 ，表示該物品爲附件， q 是所屬主件的編號）
 *
 *
 *
 */
public class ShoppingList {
    public static int getMaxValue(int[] val, int[] weight, int[] q, int n, int w) {
        int[][] dp = new int[n + 1][w + 1];
        for (int i = 1; i <= n; i++) {
            for (int j = 1; j <= w; j++) {
                if (q[i-1] == 0) {  // 主件
                    if (weight[i - 1] <= j) // 用j這麼多錢去買 i 件商品 能夠得到最大價值
                        dp[i][j] = Math.max(dp[i - 1][j], dp[i - 1][j- weight[i - 1]]+ val[i - 1]);
                }
                else { //附件
                    if (weight[i - 1] + weight[q[i - 1]] <= j) //附件的話 加上主件一塊兒算
                        dp[i][j] = Math.max(dp[i - 1][j], dp[i - 1][j- weight[i - 1]]+ val[i - 1]);
                }
            }
        }
        return dp[n][w];
    }

    public static void main(String[] args) {
        Scanner input = new Scanner(System.in);
        while (input.hasNextInt()) {
            int n = input.nextInt(); // 總錢數
            int m = input.nextInt(); // 商品個數
            int[] p = new int[m];
            int[] v = new int[m];
            int[] q = new int[m];
            for (int i = 0; i < m; i++) {
                p[i] = input.nextInt();        // 價格
                v[i] = input.nextInt() * p[i]; // 價值
                q[i] = input.nextInt();        // 主件仍是附件
            }
            System.out.println(getMaxValue(v, p, q, m, n));
        }
    }
}