P1064 金明的预算方案(洛谷)

原题传送门

在这里插入图片描述
在这里插入图片描述
思路:因为每个主件最多只有两个附件,所以枚举每个主件如果满足条件的话可以分为五种情况购买:什么都不买;只买主件;买主件和第一个附件;买主件和第二个附件;买主件和两个附件。然后分别计算出每种情况的物品价格与重要度乘积之和,最后取最大值即可.

代码参考

#include <bits/stdc++.h>
using namespace std;
int v[32001][3], p[61][3],f[32001];
/*
** v[i][j]代表第i个物品的第j个附件的价格
** p[i][j]代表第i个物品的第j个附件的重要度
** f[i]代表花了i元时,所买物品的价格与重要度乘积之和的最大值
*/
int cost2(int i,int x, int y){
    
    
     return v[i][x] + v[i][y];
}
int cost3(int i,int x, int y, int z){
    
    
     return v[i][x] + v[i][y] + v[i][z];
}
int vp(int i,int x){
    
    
     return v[i][x] * p[i][x];
}
int main() {
    
    
    int n,m;
	cin>>n>>m;
	for (int i = 1; i <= m; i++) {
    
    
		int _v, _p, _q;//每件物品的 _v:价格,_p:重要度,_q:对应的主件
		cin>>_v>>_p>>_q;
		if (!_q){
    
     // 是主件
			v[i][0] = _v;
			p[i][0] = _p;
		}else if(v[_q][1] == 0){
    
     // 是第一个附件
            v[_q][1] = _v;
            p[_q][1] = _p;
        }else{
    
     // 是第二个附件
            v[_q][2] = _v;
            p[_q][2] = _p;
        }
    }
	for(int i = 1; i <= m; i++)
		for (int j = n; j >= 0; j--) {
    
    
			if (j >= v[i][0]) // 够买主件
				f[j] = max(f[j], f[j - v[i][0]] + vp(i,0));
			if (j >= cost2(i,0, 1)) // 还够买第一个附件
				f[j] = max(f[j], f[j - cost2(i,0, 1)] + vp(i,0) + vp(i,1));
			if (j >= cost2(i,0, 2)) // 还够买第二个附件
				f[j] = max(f[j], f[j - cost2(i,0, 2)] + vp(i,0) + vp(i,2));
			if (j >= cost3(i,0, 1, 2)) // 还够买两个附件
				f[j] = max(f[j], f[j - cost3(i,0, 1, 2)] + vp(i,0) + vp(i,1) + vp(i,2));
		}
	cout<<f[n];
}

猜你喜欢

转载自blog.csdn.net/Bertil/article/details/106765005