华为机试HJ16 购物单

题目：

王强今天很开心，公司发给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 ）

示例1

输入：

1000	5
800	2	0
400	5	1
300	5	1
400	3	0
500	2	0

输出：2200

分析：

1、无后性：第i件物品是否购买的状态与预算有关、前i-1件物品最大值、是否附件、是否主件、是否可买主件+附件A、是否可买主件+附件B、是否可买主件+附件A+附件B 有关。只与前面状态有关，满足无后性要求，符合递归要求。

2、确认状态：只考虑主件，则题目与常规01背包一致。附件则用附表A、B与主件关联。

3、确认递推式：递推式如上表

4、状态数组dp：保存中间状态

5、确定边界条件：前0件物品，满意为0，对应dp[0][j] = 0;钱数为0,满意当然也为0，对应dp[i][0]=0;

 上代码

#include<iostream>
#include<algorithm>
#include "stdio.h"
using namespace std;

#define MAX 3200
//从第0个物品开始
int price[]       = {0,  80,   40,  30,  40,  50};//价格数组
int importance[]  = {0,   2,    5,   5,   3,   2};//重要度
int attributes[]  = {0,   0,    1,   1,   0,   0};//属性
int A[]           = {-1,  2,   -1,  -1,  -1,  -1};//i的附件A关联表
int B[]           = {-1,  3,   -1,  -1,  -1,  -1};//i的附件B关联表
//物品数m和钱数budget
int m = 5, budget = 100;

//i：前i个物品 j：钱数
int dp[MAX][MAX] ;

int Max(int a, int b, int c, int d, int e){
    return max(max(max(a,b), max(c,d)),e);
}

int main(){
    //初始化状态数组
    for(int i=0; i<=m; i++)
        for(int j=0; j<=budget; j++){
            //前0个物品 预算为0
            if(!i || !j) dp[i][j] = 0;
            else dp[i][j] = -1;
        }

    //递推    
    cout << maxValueRecursion(m, budget) << endl;
    //动归
    cout << maxValue(m, budget) << endl;
}

1、动归方法：递归+动态数据，传统01背包写法直接搬递推方程

/动归方式
int maxValue(int i, int j){//5 100
    printf("i = %d j=%d dp=%d\n", i, j, dp[i][j]);

    if(dp[i][j] != -1) return dp[i][j];

    //附件跳过
    if(attributes[i] != 0 ) return maxValue(i-1, j);
    if(j < price[i]){//买不起
        dp[i][j] = maxValue(i -1, j);
    }
    else  {
        int notBuy=0,main=0, a=0, b=0, ab=0;
        //只够买主件情况
        main = maxValue(i-1, j-price[i]) + price[i]*importance[i];

        //买得起不买情况
        notBuy = maxValue(i-1, j);
        printf("i,j=(%d,%d) sum=%d \n",i,j,price[i]*importance[i]);
        
        //够买主件与附件A情况
        if(A[i] != -1 && j >= price[i]+price[A[i]]) 
            a = maxValue(i-1, j-price[i]-price[A[i]]) + price[i]*importance[i] + price[A[i]]*importance[A[i]];

        //够买主件与附件B情况
        if(B[i] != -1 && j >= price[i]+price[B[i]])
            b = maxValue(i-1, j-price[i]-price[B[i]]) + price[i]*importance[i] + price[B[i]]*importance[B[i]];
        
        //够买主件与附件A 附件B 情况
        if(A[i] != -1 && B[i] != -1 && j >= price[i]+price[A[i]]+price[B[i]])
            ab =maxValue(i-1, j-price[i]-price[A[i]]-price[B[i]]) + price[i]*importance[i] + price[A[i]]*importance[A[i]] + price[B[i]]*importance[B[i]];
        
        //取最大值
        dp[i][j] = Max(notBuy, main, a, b, ab);

        printf("i,j=(%d,%d) notBuy=%d main=%d a=%d b=%d ab=%d\n",i,j, notBuy,  main,a,b,ab);
    }
    return dp[i][j];
}

2、递推方式

//递推方式
int maxValueRecursion(int m, int budget){
    for(int i = 1; i <= m; i++){//前i个物品
        for(int j = 1; j <= budget; j++){//预算为j的最大满意度
            //附件跳过 || 买不起i物品
            if(attributes[i] != 0  || price[i] > j) {
                dp[i][j] = dp[i-1][j];
                continue;
            }
            //买得起
            int notBuy=0,main=0, a=0, b=0, ab=0;
            //买主件
            main = dp[i-1][j-price[i]] + price[i]*importance[i];
            
            //不买
            notBuy = dp[i-1][ j];
            // printf("i,j=(%d,%d) sum=%d \n",i,j,price[i]*importance[i]);
            
            //买主件+附件A
            if(A[i] != -1 && j >= price[i]+price[A[i]]) 
                a = dp[i-1][ j-price[i]-price[A[i]]] + price[i]*importance[i] + price[A[i]]*importance[A[i]];

            买主件+附件B
            if(B[i] != -1 && j >= price[i]+price[B[i]])
                b = dp[i-1][ j-price[i]-price[B[i]]] + price[i]*importance[i] + price[B[i]]*importance[B[i]];

            买主件+附件A+附件B
            if(A[i] != -1 && B[i] != -1 && j >= price[i]+price[A[i]]+price[B[i]])
                ab =dp[i-1][ j-price[i]-price[A[i]]-price[B[i]]] + price[i]*importance[i] + price[A[i]]*importance[A[i]] + price[B[i]]*importance[B[i]];
            
            //同样取最大值
            dp[i][j] = Max(notBuy, main, a, b, ab);
            printf("i,j=(%d,%d) notBuy=%d main=%d a=%d b=%d ab=%d\n",i,j, notBuy,  main,a,b,ab);

        }
    }
    return dp[m][budget];
}

总结

01背包就是动态规划，确定状态，找出递推公式、边界条件后，由已知推未知，填充状态数组。若还不了解，可上中国MOOC慕课网进行深入学习。