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;
}