poj2486(树形dp)

http://poj.org/problem?id=2486

题意：一颗树，n个点（1-n），n-1条边，每个点上有一个权值，求从1出发，走V步，最多能遍历到的权值

#include <iostream>
#include <cstring>
#include <vector>
#include <cstdio>
using namespace std;
inline int max(int a,int b) {return a>b? a:b;}
const int Ni = 105;
int n,m;
bool vis[Ni];
int weg[Ni];
int dp[2][Ni][211];
vector<int> eg[Ni];
void dfs(int p)
{
    int i,j,k,son;
    vis[p]=1;
    for(i=0;i<=m;i++)
    {dp[0][p][i]=dp[1][p][i]=weg[p];}
    for(i=0;i<eg[p].size();i++)
    {
        son=eg[p][i];
        if(vis[son]) continue;
        dfs(son);
        for(j=m;j>=0;j--)
        {
            for(k=0;k<=j;k++)
            {
                //从p往son分支走得到结果回到p，人留在p
                dp[0][p][j+2] = max(dp[0][p][j+2],dp[0][p][j-k]+dp[0][son][k]);
                //从son往p走，再回到p的son,人留在son
                dp[1][p][j+2] = max(dp[1][p][j+2],dp[1][p][j-k]+dp[0][son][k]);
                //从p往son分支走得到结果
                dp[1][p][j+1] = max(dp[1][p][j+1],dp[0][p][j-k]+dp[1][son][k]);
            }
        }
    }
}
int main()
{
    while(~scanf("%d%d",&n,&m))
    {
        memset(vis,0,sizeof(vis));
        memset(dp,0,sizeof(dp));
        for(int i=1;i<=n;i++)
        {
            eg[i].clear();
            scanf("%d",weg+i);
        }
        for(int i=1;i<n;i++)
        {
            int u,v;
            scanf("%d%d",&u,&v);
            eg[u].push_back(v);
            eg[v].push_back(u);
        }
        dfs(1);
        printf("%d\n",dp[1][1][m]);
    }
    return 0;
}