NOIP提高组2006 金明的预算方案

背包升级版

附带从属关系

因为从属关系很少 所以强行把儿子绑在爸爸上dp

故必须dp俩维 dp[局数][代价] 把儿子方案们放在同一局，由上一局转移，使其不能相互影响

#include<string>
#include<cstring>
#include<cstdio>
#include<cstdlib>
#include<cmath>
#include<ctime>
#include<iostream>
#include<algorithm>

using namespace std;
int i,j,k,ans,s,n,m,l; 
int v[5000],w[5000],f[5000];
int g[5000][5];
int z[5000][5];
int p[5000];
int biao[5000][3];

int main()
{
  //freopen("budget.in","r",stdin);
 // freopen("budget.out","w",stdout);
  int a,b,c;
  scanf("%d%d",&m,&n);
  m=m/10;
  for(i=1;i<=n;i++)
  {
     scanf("%d%d%d",&a,&b,&p[i]);
     w[i]=a/10;
     v[i]=w[i]*b;
  }
  
  
  for(i=1;i<=n;i++)
  {
     if (p[i]==0) 
     {
        g[i][1]=v[i];    z[i][1]=w[i];      
        g[i][2]=v[i];     z[i][2]=w[i];               
        g[i][3]=v[i];     z[i][3]=w[i];         
        g[i][4]=v[i];     z[i][4]=w[i];      
     
     }
     if (p[i]!=0)
        { k=1;
          if(biao[p[i]][1]!=0)
              k++;
          biao[p[i]][k]=i;    
        }

  }
  for(i=1;i<=n;i++)
  {
     if (p[i]==0) 
     {
        if(biao[i][1]!=0)   
            { if (biao[i][2]!=0)
                {
                   g[i][3]+=v[biao[i][2]];
                   z[i][3]+=w[biao[i][2]];
                   
                   g[i][4]+=v[biao[i][2]];
                   z[i][4]+=w[biao[i][2]];
                   
                }
                  g[i][2]+=v[biao[i][1]];
                   z[i][2]+=w[biao[i][1]];
                   
                   g[i][4]+=v[biao[i][1]];
                   z[i][4]+=w[biao[i][1]];
                
                
             }
  }
  }
  
  for(i=1;i<=n;i++)
  {
     if(p[i]!=0) continue;
     
     for(j=m;j>=w[i];j--)
       for(k=1;k<=4;k++)
         {
            if(j>=z[i][k])  f[j]=max(f[j],f[j-z[i][k]]+g[i][k]);               
         
         }                
                   
                   
  
  }
  
      
    
   cout<<f[m]*10;
   
   
   return 0;
}