hdu 1561 树形dp

来源:互联网 发布:mac隐藏菜单栏 编辑:程序博客网 时间:2024/06/06 01:46



链接:戳这里


The more, The Better
Time Limit: 6000/2000 MS (Java/Others)    Memory Limit: 32768/32768 K (Java/Others)

Problem Description
ACboy很喜欢玩一种战略游戏,在一个地图上,有N座城堡,每座城堡都有一定的宝物,在每次游戏中ACboy允许攻克M个城堡并获得里面的宝物。但由于地理位置原因,有些城堡不能直接攻克,要攻克这些城堡必须先攻克其他某一个特定的城堡。你能帮ACboy算出要获得尽量多的宝物应该攻克哪M个城堡吗?
 
Input
每个测试实例首先包括2个整数,N,M.(1 <= M <= N <= 200);在接下来的N行里,每行包括2个整数,a,b. 在第 i 行,a 代表要攻克第 i 个城堡必须先攻克第 a 个城堡,如果 a = 0 则代表可以直接攻克第 i 个城堡。b 代表第 i 个城堡的宝物数量, b >= 0。当N = 0, M = 0输入结束。
 
Output
对于每个测试实例,输出一个整数,代表ACboy攻克M个城堡所获得的最多宝物的数量。
 
Sample Input
3 2
0 1
0 2
0 3
7 4
2 2
0 1
0 4
2 1
7 1
7 6
2 2
0 0
 
Sample Output
5
13
 

思路:

设置dp[i][j] 表示以i作为根节点,向下取j个节点的最优值

当前根节点为u,其中一个儿子为v

两层for枚举  当前父亲u还能分配j个节点,分配给v儿子k个,那么其他的儿子(包括父亲本身)总共就是(j-k)个

那么儿子v至少分配一个,但是最多分配(num-1)个 ,父亲占去一个


代码:

#include<iostream>#include<cstdio>#include<cstring>#include<algorithm>#include<string>#include<vector>#include <ctime>#include<queue>#include<set>#include<map>#include<stack>#include<iomanip>#include<cmath>#define mst(ss,b) memset((ss),(b),sizeof(ss))#define maxn 0x3f3f3f3f#define MAX 1000100///#pragma comment(linker, "/STACK:102400000,102400000")typedef long long ll;typedef unsigned long long ull;#define INF (1ll<<60)-1using namespace std;int n,m;struct edge{    int v,next;}e[10010];int head[220],a[220],tot;void Add(int u,int v){    e[tot].v=v;    e[tot].next=head[u];    head[u]=tot++;}int dp[220][220];void DFS(int u,int fa,int num){    dp[u][1]=a[u];    for(int i=head[u];i!=-1;i=e[i].next){        int v=e[i].v;        if(v==fa) continue;        if(num>1) DFS(v,u,num-1);        ///printf("%d %d %d\n",u,v,num);        for(int j=num;j>=1;j--){            for(int k=1;k<j;k++){                dp[u][j]=max(dp[u][j],dp[v][k]+dp[u][j-k]);            }        }    }}int main(){    while(scanf("%d%d",&n,&m)!=EOF){        if(n==0 && m==0) break;        mst(dp,0);        mst(a,0);        mst(head,-1);        tot=0;        for(int u=1;u<=n;u++){            int v;            scanf("%d%d",&v,&a[u]);            Add(v,u);            Add(u,v);        }        DFS(0,-1,m+1);        printf("%d\n",dp[0][m+1]);    }    return 0;}


0 0
原创粉丝点击