华为oj--中级--购物单

package azr.E001.购物单;


/**
 * 描述: 王强今天很开心，公司发给N元的年终奖。王强决定把年终奖用于购物，他把想买的物品分为两类：主件与附件，附件是从属于某个主件的，下表就是一些主件与附件的例子：
 *        主件 附件
 *        电脑 打印机，扫描仪
 *        书柜 图书
 *        书桌 台灯，文具
 *        工作椅 无
 *      如果要买归类为附件的物品，必须先买该附件所属的主件。每个主件可以有 0 个、 1 个或 2 个附件。附件不再有从属于自己的附件。
 *      王强想买的东西很多，为了不超出预算，他把每件物品规定了一个重要度，分为 5 等：用整数 1 ~ 5 表示，第 5 等最重要。
 *      他还从因特网上查到了每件物品的价格（都是 10 元的整数倍）。他希望在不超过 N 元（可以等于 N 元）的前提下，使每件物品的价格与重要度的乘积的总和最大。
 *         设第 j 件物品的价格为 v[j] ，重要度为 w[j] ，共选中了 k 件物品，编号依次为 j 1 ， j 2 ，……， j k ，则所求的总和为：
 *         v[j 1 ]*w[j 1 ]+v[j 2 ]*w[j 2 ]+ … +v[j k ]*w[j k ] 。（其中 * 为乘号）
 *         请你帮助王强设计一个满足要求的购物单。
 * 
 * 输入: 输入的第 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 是所属主件的编号）
 * 输出: 输出文件只有一个正整数，为不超过总钱数的物品的价格与重要度乘积的总和的最大值（ <200000 ）。
 * 
 * 样例输入: 1000 5 
 *         800 2 0 
 *         400 5 1 
 *         300 5 1 
 *         400 3 0 
 *         500 2 0
 * 样例输出: 2200
 */


import java.util.Scanner;


public class Main {


public static void main(String[] args) {

Scanner in = new Scanner(System.in);
int N = in.nextInt();// 总钱数
int m = in.nextInt();// 希望购买物品个数


int[] v = new int[m];
int[] p = new int[m];
int[] q = new int[m];

for(int i = 0; i < m; i++) {
v[i] = in.nextInt();
p[i] = in.nextInt();
q[i] = in.nextInt();
}

in.close();

// int N = 10;
// int m = 5;
//
// int[] v = {8, 4, 3, 4, 5};
// int[] p = {2, 5, 5, 3, 2};
// int[] q = {0, 1, 1, 0, 0};

System.out.println(getMaxValue(v, p, q, m, N));
}


private static int getMaxValue(int[] v, int[] p, int[] q, int m, int n) {


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

// 处理主件，使用01背包 动态规划。dp为价格乘重要度，flag中true表示有该物品，false表示没有。
        for(int i = m - 1; i >= 0; i--) {
        if(q[i] == 0) { // 主件
        for(int j = 0; j <= n; j++) {
                if(v[i] == j) {         
    dp[i][j] = v[i] * p[i];
    flag[i][j] = true;
    }
    if(i == m - 1){
    if(j >= v[i]) {
    dp[i][j] = v[i] * p[i];
    flag[i][j] = true;
    }
    } else {
    int t1 = 0;
    int t2 = 0;
    for(int k = i; k < m; k++) { // 01背包问题，去掉附件干扰
    if(q[k + 1] == 0) {
    t1 = dp[k + 1][j];
    if(j - v[i] >= 0) {
        t2 = dp[k + 1][j - v[i]] + v[i] * p[i];
        }
        dp[i][j] = Math.max(t1, t2);
        if(dp[i][j] != 0) {
        flag[i][j] = dp[i][j] == t2 ? true : false;
        }
        break;
    }
    }
    }
                }
        } 
        }
        
        int max = 0;
        for(int i = 0; i < dp.length; i++) {
        for(int j = 0; j < dp[i].length; j++) {
        if(dp[i][j] > max) {
        max = dp[i][j];
        }
        }
        }
        
        // 处理附件
        for(int i = m - 1; i >= 0; i--) {
        if(q[i] != 0) { // 附件
        for(int j = 0; j <= n; j++) {
       
        if(flag[q[i] - 1][j] == true) { // 购买了附件对应的主件
        int sy = n - v[q[i] - 1]; // sy 为除去购买主件后剩余的金额
        if(sy > v[i - 1]) {
        max = max > v[q[i] - 1] * p[q[i] - 1] + v[i] * p[i] ? max : v[q[i] - 1] * p[q[i] - 1] + v[i] * p[i];
        }
        } else { // 未购买附件对应的主件
       
        }
       


                }
        } 
        }
        
        return max;
}


}