快谈混合背包问题

#include<iostream>
using namespace std;
int v[1005],w[1005],s[1005];//s[i]表示物品特性，是01还是多重还是完全
int n,m;
int dp[10005],vis[10005];//vis[i]==1表示第i个物品是完全背包，==0表示01背包
int v1[10005],w1[10005];//最终转换成的01背包和完全背包
int main(){
	int index=0;
	cin>>n>>m;
	for(int i=1;i<=n;i++){
		cin>>v[i]>>w[i]>>s[i];
	}
	for(int i=1;i<=n;i++){
		if(s[i]==0){//完全背包问题
		    vis[++index]=1;
			v1[index]=v[i];
			w1[index]=w[i];
			continue;
		}
		if(s[i]==-1){//01背包问题
			v1[++index]=v[i];
			w1[index]=w[i];
			continue;
		}
		//多重背包问题
		for(int j=1;j<=s[i];j<<=1){
			s[i]-=j;
			v1[++index]=j*v[i];
			w1[index]=j*w[i];
		}
		if(s[i]){
			v1[++index]=s[i]*v[i];
			w1[index]=s[i]*w[i];
			s[i]=0;
		}
	}
	for(int i=1;i<=index;i++){
		if(vis[i]==1){//完全背包
			for(int j=v1[i];j<=m;j++){
				dp[j]=max(dp[j],dp[j-v1[i]]+w1[i]);
			}
		}else{//01背包
			for(int j=m;j>=v1[i];j--){
				dp[j]=max(dp[j],dp[j-v1[i]]+w1[i]);
			}
		}
	}
	cout<<dp[m]<<endl;
	return 0;
}