POJ2486【树形DP】

题意：

给出一棵树，然后从1开始走，问你走k步所能遍历到的最大权值和.

思路：

我们要求从1这个节点走K步所获得的最大值，所以dp就很容易想到>>>>>dp[node][k]: 代表从这个节点走k步的最大值。
但是在状态转移的时候，好像不能很清楚的从子节点通过某种关系到父节点。
所以套路：考虑从某个节点他最终回不回到本身。
那么转移就很容易了。
dp[i][j][0]代表从节点i走j步回到i的最大值/dp[i][j][1]代表从节点i走j步不回到i的最大值
[son] [k - 2] [0] + dp[father] [j - k] [0] => [father] [ j ] [0]
[son] [k - 2] [0] + dp[father] [j - k] [1] => [father] [ j ] [1]
[son] [k - 1] [1] + dp[father] [j - k] [0] => [father] [ j ] [1]
现在转移有了，初始化呢？
每个节点走0步回到本身: val[node];
意味着所有节点不管走多少步，最小就是本身的价值。
PS：
转移中有一个小地方，自己一开始还疑惑

  for(int j=k;j>=1;j--){      for(p=1;p<=j;p++){       }  }

凭什么我里面要调用步数少的节点，却不先更新步数小的而先更新大的？
类似0/1背包，因为我每次对于这个子节点我只能用一次，
如果先枚举小的，又取小的，这个子节点不是只取了一次。
那么我对于步数小的那些dp值就是初始值？
肯定不是啊，那些步数小的dp值在其他son节点中更新了呀。

Code:

const int N=1e2+10;struct Edge{    int v;    int Next;}edge[N<<1];int head[N],tol;int n,k,val[N];int dp[N][N<<1][2];void init(){    tol=0;    memset(head,-1,sizeof(head));    memset(dp,0,sizeof(dp));}void add(int u,int v){    edge[tol].v=v;    edge[tol].Next=head[u];    head[u]=tol++;}bool vis[N];void DFS(int u){    int j,p;    for(int i=head[u];~i;i=edge[i].Next){        int v=edge[i].v;        if(vis[v]) continue;        vis[v]=true;        DFS(v);        for(int j=k;j>=1;j--){                for(p=1;p<=j;p++){                    if(p>=2){                        dp[u][j][0]=max(dp[u][j][0],dp[v][p-2][0]+dp[u][j-p][0]);                        dp[u][j][1]=max(dp[u][j][1],dp[v][p-2][0]+dp[u][j-p][1]);                    }                    dp[u][j][1]=max(dp[u][j][1],dp[v][p-1][1]+dp[u][j-p][0]);                }        }    }}int main(){    int u,v;    while(~scanf("%d%d",&n,&k)){        init();        for(int i=1;i<=n;i++)        {            scanf("%d",&val[i]);            for(int j=0;j<=k;j++)                dp[i][j][0]=dp[i][j][1]=val[i];        }        for(int i=1;i<n;i++){            scanf("%d%d",&u,&v);            add(u,v);            add(v,u);        }        memset(vis,false,sizeof(vis));        vis[1]=true;        DFS(1);        printf("%d\n",max(dp[1][k][0],dp[1][k][1]));    }    return 0;}