您的位置:首页 > 其它

POJ1155-树形DP&&背包

2017-06-27 21:32 495 查看
技巧:num数组,用来记录在此树上最多选择多少个。其他就是背包了

#include <cstdlib>
#include <cstdio>
#include <iostream>
#include <cstring>
using namespace std;
const int maxn=3002;
int m,n;
int len;
int dp[maxn][maxn],tem[maxn],a[maxn][maxn],head[maxn],num[maxn];
struct node
{  int now;
int val;
int next;

}tree[9005];
void add(int a,int b,int c)
{  tree[len].now=a;
tree[len].val=b;
tree[len].next=head[c];//建立一个树,链式前进星,
head[c]=len++;

}
int dfs(int root)
{   int p;
for(int i=head[root];i!=-1;i=tree[i].next)
{   p=tree[i].now;
dfs(p);
for(int l=0;l<=num[root];l++)
tem[l]=dp[root][l];
for(int k=0;k<=num[root];k++)
for(int j=1;j<=num[p];j++)
{     dp[root][k+j]=max(dp[root][j+k],dp[p][j]+tem[k]-tree[i].val);

}

num[root]+=num[p];
}

return 0;
}
int main()
{
int a,b;
int k;
while(~scanf("%d%d",&n,&m))
{   len=0;
memset(head,-1,sizeof(head));
for(int i=1;i<=n-m;i++)
{   scanf("%d",&k);
num[i]=0;
for(int c=0;c<k;c++)
{scanf("%d%d",&a,&b);
add(a,b,i);}
}

for(int i=1;i<=n;i++)
for(int j=1;j<=m;j++)
dp[i][j]=-9999999;
for(int i=n-m+1;i<=n;i++)
{scanf("%d",&dp[i][1]);
num[i]=1;}
dfs(1);
//for(int i=1;i<=n-m;i++)
// printf("%d  ",head[i]);
//printf("^^^^\n");
for(int i=m;i>=0;i--)
{    //printf("%d  ",dp[i][1]);
if(dp[1][i]>=0)
{printf("%d\n",i);break;}
}

}

return 0;
}
内容来自用户分享和网络整理,不保证内容的准确性,如有侵权内容,可联系管理员处理 点击这里给我发消息
标签: