洛谷P1064金明的預算方案

題目傳送門

題目含義：總錢數m，物品總件數n。每個物品只能買一次，想要買附件必須買主件。求不超過總錢數的最大價值。

主件1與附件2、3、4的買法組合有：12、13、123、14、124、134、1234，可以把這些組合看作一個整體，每次從中選取一個組合。這是一個01揹包+分組揹包的問題。

算法的時間複雜度爲：

下面是代碼：

#include<iostream>
#include<vector>
using namespace std;
int m,n;
int f[40000];
vector<pair<int,int>>item[65];
int main(){
	cin>>m>>n;
	for(int i=1;i<=n;i++){
		int w,v,q;
		cin>>w>>v>>q;
		if(q==0){
			item[i].push_back({w,w*v});
		}else{
			int sz=item[q].size();
			for(int j=0;j<sz;j++)
				item[q].push_back({item[q][j].first+w,item[q][j].second+w*v});
		}
	}
	for(int i=1;i<=n;i++){
		if(item[i].empty())continue;
		for(int j=m;j>=1;j--)
			for(int k=0;k<item[i].size();k++)
				if(j>=item[i][k].first){
					f[j]=max(f[j],f[j-item[i][k].first]+item[i][k].second);
				}
	}
	cout<<f[m];
	return 0;
}