hdu 1561（树形DP）

hdu 1561

题目：http://acm.hdu.edu.cn/showproblem.php?pid=1561

先开始建图，发现可能成环，要先把所有的环都去掉，新建一个节点0，把a==0的点全都连到0上，然后就是root = 0 的一棵树。对每一个节点的所有儿子节点进行背包。后来看了网上的代码，发现都没考虑成环的情况，不知道是我想多了，本来就不需要考虑，还是很多人都没注意到。关于如何找出所有成环的点，先开始还没想到，问了一下，原来走一遍强连通接好了 = = 。后来我也写了个没判定成环的，果然是AC了。代码如下，d[i][j]表示以 i 为父亲节点，已攻克了 j 个城堡能获得最大值。状态转移方程：d[u][i] = max(d[u][i],d[u][j]+d[v][m-j]) ，j<i ，01背包注意循环顺序及初始化。为了方便，多了个0节点，所以m++。

代码如下：

#include<cstdio>#include<cstring>#include<algorithm>#include<vector>using namespace std;const int MAXN = 222 ;vector <int> G[MAXN];int val[MAXN];int n,m;int d[MAXN][MAXN];void dfs(int u){    for(int i=0;i<=m;i++)        d[u][i]=0;    d[u][1]=val[u];    for(int i=0;i<G[u].size();i++)    {        int v = G[u][i];        //printf("u = %d,v = %d\n",u,v);        dfs(v);        for(int j=m;j>=1;j--)        {            for(int k=0;k<j;k++)            {                d[u][j]=max(d[u][j],d[u][j-k]+d[v][k]);            }        }    }}int main(){    while(~scanf("%d%d",&n,&m))    {        if(n+m==0) break;        for(int i=0;i<=n;i++)            G[i].clear();        int a,b;        for(int i=1;i<=n;i++)        {            scanf("%d%d",&a,&b);            if(a>0)                G[a].push_back(i);            else G[0].push_back(i);            val[i]=b;        }        val[0]=0;        m++;        dfs(0);        printf("%d\n",d[0][m]);    }    return 0;}