luogu 1064 金明的预算方案

01背包 变形,有主附件的背包内则更改决策

original: 1) 不选   2)选，f[j-w[i]]+v[i]

now :     1)不选   2)选主   3)主 附1  4)主 附2   5)主 附1 附2

遍历 i 时跳过附件即可

#include<iostream>
using namespace std;

int m,n,vv,pp,qq;
int w[][],v[][],cnt[];
int f[];

int main()
{
    cin>>m>>n;
    for(int i=;i<=n;i++){
        cin>>vv>>pp>>qq;
        if(!qq){
            w[i][]=vv;
            v[i][]=vv*pp;
        }else{
            cnt[qq]++;
            w[qq][cnt[qq]]=vv;
            v[qq][cnt[qq]]=vv*pp;
        }
    }
    for(int i=;i<=n;i++)
    for(int j=m;j>=w[i][];j--){

        f[j]=max(f[j],f[j-w[i][]]+v[i][]);  

        if(j>=w[i][]+w[i][])
           f[j]=max(f[j],f[j-w[i][]-w[i][]]+v[i][]+v[i][]);  

        if(j>=w[i][]+w[i][])
           f[j]=max(f[j],f[j-w[i][]-w[i][]]+v[i][]+v[i][]);    

        if(j>=w[i][]+w[i][]+w[i][])
           f[j]=max(f[j],f[j-w[i][]-w[i][]-w[i][]]+v[i][]+v[i][]+v[i][]);
    }
    cout<<f[m]<<endl;
    return ;
}