金明的预算方案

题目
分析一下，若想选附件，必然要选其主件，看上去是个依赖背包问题，也就是树形DP，但是这个题目限制了一个问题，也就是一个主件至多有2个附件，那么也就只有4种方案，只选主件，选主件和附件1，选主件和附件2，选主件和附件1和附件2。只有4种方案，所以将其转化成为一个组合背包问题。

#include <cstdio>
#include <iostream>
#include <cstring>
using namespace std;
int f[40000],w[40000],v[40000],p[40000],s[40000][3];
int main()
{
    int n,m,k=0;

    scanf("%d%d",&m,&n);

    for(int i=1;i<=n;i++)
     {
        scanf("%d%d%d",&w[i],&v[i],&p[i]);
        v[i]=v[i]*w[i];
        if(p[i]!=0) s[p[i]][++s[p[i]][0]]=i;
     }

    for(int i=1;i<=n;i++)
     if(p[i]==0)
      for(int j=m;j>=0;j--)
      {
        if(j-w[i]>=0)
         f[j]=max(f[j],f[j-w[i]]+v[i]);
        if(j-w[i]-w[s[i][1]]>=0&&s[i][1]!=0)
         f[j]=max(f[j],f[j-w[i]-w[s[i][1]]]+v[i]+v[s[i][1]]);
        if(j-w[i]-w[s[i][2]]>=0&&s[i][2]!=0)
         f[j]=max(f[j],f[j-w[i]-w[s[i][2]]]+v[i]+v[s[i][2]]);
        if((j-w[i]-w[s[i][1]]-w[s[i][2]])>=0&&s[i][0]==2)
         f[j]=max(f[j],f[j-w[i]-w[s[i][1]]-w[s[i][2]]]+v[i]+v[s[i][1]]+v[s[i][2]]);
      }
    printf("%d",f[m]);

    return 0;
}