【牛客华为机试】购物单

一、题目描述

  公司发给王强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.输入描述:

  输入的第 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 是所属主件的编号）

2.输出描述:

  输出文件只有一个正整数，为不超过总钱数的物品的价格与重要度乘积的总和的最大值（ <200000 ）。

二、题目分析

1.分析

  题目属于典型的背包问题，将物品信息封装后存入数组，每次挑出一件物品时，我们面临购买和不购买两个选择。如果购买，则除去购买当前物品后剩余的钱：N-v[j] ，所能买到的物品应该也满足买到的物品和重要乘积最大；如果不够买，则和挑选上一件物品 j-1 时情况相同。所以每次选择物品时，我们都要用当前物品前 j-1件物品的信息，典型的需要使用动态规划求解。
  我们设f(n,m)表示使用钱数n，在挑选（买或者不买）第m件物品时，能够买到的物品的价格和重要性乘积最大值。则：
  1.子问题。对于第j件物品我们需要求出f(n,j)。如果不挑选此物品，则：f(n,j) = f(n,j-1)；如果挑选当前物品，则：f(n,j) = max[ f(n,j), f(n-vj, j-1) + vj* wj ]
  2.临界条件。当前物品需为主件，附件默认不购买；对于第j件物品如果购买，则钱数必须大于当前物品单价。
  3.转移方程。不购买时f(n,j) = f(n,j-1)；购买时 f(n,j) = max[ f(n,j), f(n-vj, j-1) + vj* wj ]

2.代码

import java.util.Scanner;

public class Main{
   
    public static void main(String[] args) {
   
        Scanner scanner = new Scanner(System.in);
        while (scanner.hasNext()) {
   
            int N = scanner.nextInt();
            int n = scanner.nextInt();
            Good[] goods = new Good[n];
            for (int i = 0; i < n; i++) {
   
                int v = scanner.nextInt();
                int p = scanner.nextInt();
                int q = scanner.nextInt();
                // 读取到当前物品为附件
                if (q>0){
   
                	// q的值为当前附件所属主件的编号，所以对应的主件索引为q-1
                	// 若对应的主件不存在第一个附件，则将当前附件赋值给第一个附件
                    if (goods[q-1].annex1 == -1) {
   
                        goods[q-1].annex1 = i;
                    } else {
   
                    	// 若对应的主件存在第一个附件，则将当前附件赋值给第二个附件
                        goods[q-1].annex2 = i;
                    }
                }
                goods[i] = new Good(v,p,q);
            }
            findMaxValuePriority(N,goods);
        }
    }

    private static void findMaxValuePriority(int N, Good[] goods){
   
    	// 相当于f(n,m)函数，记录在挑选第j件物品时，能买到的价格乘重要性的最大值
        int[][] vps = new int[N+1][goods.length+1];
        for (int j = 1; j <= goods.length; j++) {
   
            // 计算当前商品的主附件价格和重要性信息
            Good curr = goods[