hdu 1561 The more, The Better（树形dp）

来源：互联网发布：ssd优化编辑：程序博客网时间：2024/05/29 19:34

dp[root][i]root代表这是第几个节点，i代表在root这个节点攻陷的城市数量

#include<stdio.h>#include<iostream>#include<string.h>#include<algorithm>#define maxn 1000010using namespace std;int dp[210][210],fa[210],bro[210],son[210],c[210];int n,p;void dfs(int root){    int k=son[root];    dp[root][1]=c[root];    while(k)    {        dfs(k);        for(int i=p;i>=1;i--)        {            int q=i;            if(root==0)q=i+1;//如果节点为0，特殊处理，0和其他节点的边不能计算进去            for(int j=0;j<q;j++)            {                dp[root][i]=max(dp[root][i-j]+dp[k][j],dp[root][i]);            }        }        k=bro[k];    }}int main(){    int a;while(~scanf("%d%d",&n,&p)&&n){memset(fa,0,sizeof(fa));memset(bro,0,sizeof(bro));memset(son,0,sizeof(son));memset(dp,0,sizeof(dp));c[0]=0;for(int i=1;i<=n;i++){    scanf("%d%d",&a,&c[i]);    bro[i]=son[a];    fa[i]=a;    son[a]=i;}dfs(0);int ans=dp[0][p];printf("%d\n",ans);}    return 0;}

0 0