洛谷-动态规划引入-P1064-金明的预算方案

[NOIP2006 提高组] 金明的预算方案

题目描述

金明今天很开心，家里购置的新房就要领钥匙了，新房里有一间金明自己专用的很宽敞的房间。更让他高兴的是，妈妈昨天对他说：“你的房间需要购买哪些物品，怎么布置，你说了算，只要不超过 $n$ 元钱就行”。今天一早，金明就开始做预算了，他把想买的物品分为两类：主件与附件，附件是从属于某个主件的，下表就是一些主件与附件的例子：

主件	附件
电脑	打印机，扫描仪
书柜	图书
书桌	台灯，文具
工作椅	无

如果要买归类为附件的物品，必须先买该附件所属的主件。每个主件可以有 $0$ 个、$1$ 个或 $2$ 个附件。每个附件对应一个主件，附件不再有从属于自己的附件。金明想买的东西很多，肯定会超过妈妈限定的 $n$ 元。于是，他把每件物品规定了一个重要度，分为 $5$ 等：用整数 $1\sim 5$ 表示，第 $5$ 等最重要。他还从因特网上查到了每件物品的价格（都是 $10$ 元的整数倍）。他希望在不超过 $n$ 元的前提下，使每件物品的价格与重要度的乘积的总和最大。

设第 $j$ 件物品的价格为 $v_j$，重要度为$w_j$，共选中了 $k$ 件物品，编号依次为 $j_1,j_2,\dots ,j_k$，则所求的总和为：

$v_{j_1}\times w_{j_1}+v_{j_2}\times w_{j_2}+\cdots +v_{j_k}\times w_{j_k}$。

请你帮助金明设计一个满足要求的购物单。

输入格式

第一行有两个整数，分别表示总钱数 $n$ 和希望购买的物品个数 $m$。

第 $2$ 到第 $(m+1)$ 行，每行三个整数，第 $(i+1)$ 行的整数 $v_i$，$p_i$，$q_i$ 分别表示第 $i$ 件物品的价格、重要度以及它对应的的主件。如果 $q_i=0$，表示该物品本身是主件。

输出格式

输出一行一个整数表示答案。

样例 #1

样例输入 #1

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

样例输出 #1

2200

提示

数据规模与约定

对于全部的测试点，保证 $1\le n\le 3.2\times 10^4$，$1\le m\le 60$，$0\le v_i\le 10^4$，$1\le p_i\le 5$，$0\le q_i\le m$，答案不超过 $2\times 10^5$。

NOIP 2006 提高组 第二题

举例

首先这道题是一个背包问题
我们先举个例字

物品	价格	重要度	主件
物品1	4	3	0
物品2	3	3	0
物品3	4	4	2
物品4	5	5	0
物品5	3	5	2

这是五个物品的价格，重要度，和附属（主件为0代表没有附属，主件为 i 就是物品 i 的附属）

现我们需要将主件以及其附属物品排序在一起为一个组合

组合	花费	乘积	附属
Class1	4	12	0
Class2	3	9	0
Class2	4	16	1 (组合Class2的子一个附属,也就是物品3）
Class2	3	15	2 (组合Class2的子一个附属,也就是物品5）
Class3	5	25	0

花费	1	2	3	4	5	6	7	8	9	10
Class1	0	0	0	12	12	12	12	12	12	12
Class2	0	0	9	12	12	24	25	25	25	40
Class3	0	0	9	12	25	25	25	34	37	40

从这个表格我们可以先用二维数组的思路来进行思考，首先我们以组合的种类进行表示 i ，用 钱 来表示 j，首先是 i=1，也就是Class1(用c1表示），c1的花费为4，也就是在 j=4时我们可以填入数组中的值为它的乘积也就是12，由于c1没有附属所以从 j=4 到 j=10 都是12；（但是在实际代码中，我们写 j 时是不能从前往后判断的，我们需要从 后往前判断，也就是从 j = 10 来开始判断，因为二维数组的值初始为0 ,根据下边的这个dp方程

//v为花费，s为乘积
dp[j] = max(dp[j],dp[j-v]+s);

 如果我们从前向后看当  j=4 时 dp[4] = 12 ; 当 j = 8时 dp[8-4]+s=dp[4]+s=12+12=24,显然与我们预期的不符，因为运算重复了，
 但是如果我们从后往前看，dp[10]=dp[10-4]+s=dp[6]+s=0+12=12, dp[9]=dp[9-4]+s=dp[5]+s=0+12=12;这就是为什么 j 要从后向前遍历

然后我们再看c2，c2的价格为3，从 j = 10 开始 ，正好c2 的所有加一起为10，所以dp[10] = 9+16+15 = 40, j = 9 时 是 c2的主件和附属1最大所以是 9+16 =25，当 j = 6时就只能是 c2主件和附属2 ，9+15 = 24，当 j=5 时因为主件 9 没有第一个主件 12 大所以dp[5]=12, 直到 j=3时，dp[3] = 9;
从c2的j=5时的判断可以让我们的二维数组用一维表示，所以其实就是一个一直判断并且覆盖的问题。

#include<bits/stdc++.h>
using namespace  std;
//用结构体来存 id表示类，v是价格，s是乘积，f是附属编号
struct node{
    int id,v,s,f;
}a[32005];
// b用来存附属的个数（也就是说b数组的 下标代表的是 类，其中的值代表的是个数
//dp的下标表示价钱，其中的值是乘积
int b[32005],dp[32005];

bool cmp(node x,node y){
    if(x.id==y.id) 
        return x.f<y.f;
    return x.id<y.id;
}
int main(){
    int n,m,v,p,q;
    cin>>n>>m;
    for(int i=1;i<=m;i++){
        cin>>v>>p>>q;
        //判断是主件还是附件
        if(q==0) a[i].id=i,a[i].v=v,a[i].s=v*p,a[i].f=0;
        else a[i].id=q,a[i].v=v,a[i].s=v*p,a[i].f=++b[q];
    }
    //对 类 和 附件 进行排序
    sort(a+1,a+1+m,cmp);
	// i 表示对所有物品进行遍历
    for(int i=1;i<=m;i++){
    	//只对主件判断
        if(a[i].f) continue;
      
        for(int j=n;j>=a[i].v;j--){
        	//这个就是上边所说的要从后向前遍历
             dp[j] = max(dp[j],dp[j-a[i].v]+a[i].s);
             //主件判断完之后，判断附件一个主件。只可能有0，1，2个附件。因为之前对类和附件用sort排序了，所以可以直接 i+1，i+2 判断
            
             if(a[i+1].id==a[i].id&&j>=a[i].v+a[i+1].v){
                dp[j] = max(dp[j],dp[j-a[i].v-a[i+1].v]+a[i].s+a[i+1].s);
             }
             if(a[i+2].id==a[i].id&&j>=a[i].v+a[i+2].v){
                dp[j] = max(dp[j],dp[j-a[i].v-a[i+2].v]+a[i].s+a[i+2].s);
             }
             // 上边的 i+1 和 i+2 比较 当只能选一个附件 哪一个更大
             if(a[i+2].id==a[i].id&&j>=a[i].v+a[i+2].v+a[i+1].v){
                dp[j] = max(dp[j],dp[j-a[i].v-a[i+2].v-a[i+1].v]+a[i].s+a[i+2].s+a[i+1].s);
             }

        }
    }
    cout<<dp[n];
    return 0;

}