洛谷P1064 金明的预算方案

//依赖关系最多2种 可以用分组背包水过  
#include<bits/stdc++.h>
using namespace std;
#define  w first
#define  v second
const int maxv=32005;
const int maxn=65;
int n,m,k,tot,f[maxv],ID[maxn];
vector< pair<int,int> >d[maxn];
int main()
{
    scanf("%d%d",&m,&n);
    for(int i=1,x,y,z;i<=n;++i) 
    {
        scanf("%d%d%d",&x,&y,&z);//坑 注意审题 给出编号按输入顺序 
        if(!z) d[++tot].push_back(make_pair(x,y)),ID[i]=tot; 
        else d[ID[z]].push_back(make_pair(x,y)),ID[i]=ID[z];
    }
    for(int i=1;i<=tot;++i)
    {
        d[i][0]=make_pair(d[i][0].w,d[i][0].w*d[i][0].v);
        //直接把价值预处理成重要度乘体积 注意处理顺序 
        if(d[i].size()==2) d[i][1]=make_pair(d[i][1].w+d[i][0].w,d[i][1].v*d[i][1].w+d[i][0].v);
        //一个附件 看成两组——只有主件/主件+附件 
        if(d[i].size()==3)//两个附件 看成四组——只有主件/主件+附件1/主件+附件2/主件+附件1+附件2 
        {
            d[i].push_back(make_pair(d[i][0].w+d[i][1].w+d[i][2].w,d[i][0].v+d[i][1].v*d[i][1].w+d[i][2].v*d[i][2].w));
            d[i][1]=make_pair(d[i][1].w+d[i][0].w,d[i][1].v*d[i][1].w+d[i][0].v);
            d[i][2]=make_pair(d[i][2].w+d[i][0].w,d[i][2].v*d[i][2].w+d[i][0].v);
        }
    }
    for(int k=1;k<=tot;++k)//分组背包 
        for(int j=m;j>=0;--j)
            for(int i=0;i<d[k].size();++i)
                if(j>=d[k][i].w) f[j]=max(f[j],f[j-d[k][i].w]+d[k][i].v);
    printf("%d",f[m]);
    return 0;
}