POJ2486Apple tree题解

题目地址：Apple tree
题目大意：
有一个树，n个节点，第一个节点为根节点，每个点都有一个权值，每个点都可以移动到相邻的点，给你树的连接情况，求出，走k步最多获得多少权值？
输入（多数据，EOF结束）：
n，k
w1 w2 w3……wn
以下n-1行，表示哪两个点相连。
输出：
最大值
思路：简单的树形DP

#include 
#include 
#include 
#include 
#include 
#include 
#include 
#include 
#include 
#include 
#include 
#include 
#include 
#include 
#include 
#include 
#include 
#include 
#include 
#include 
#include 
#include 
#include 
#include 
#include 
#include 
#include 
#include 
#include 
#include 
#include 
#include 
#include 
#include 
#include 
#include 
#include 
#include 
#include 
#include 
#include 
#include 
#include 
#include 
#include 
#include 
#include 
using namespace std;
vector<int> a[201];
int m,f0[201][201],f1[201][201],v[201],p[201],n;
void dfs(int x){
    int i,j,k,d;
    for(i=0;i<=m;i++)
        f0[x][i]=f1[x][i]=v[x];
    p[x]=1;
    for(i=0;iif(p[d])continue;
        dfs(d);
        for(j=m;j>=0;j--)
            for(k=0;k<=j;k++){
                f0[x][j+2]=max(f0[x][j+2],f0[x][j-k]+f0[d][k]);
                f1[x][j+2]=max(f1[x][j+2],f1[x][j-k]+f0[d][k]);
                f1[x][j+1]=max(f1[x][j+1],f0[x][j-k]+f1[d][k]);
            }
    }
}
int read(){
    char c=getchar();
    int k=0;
    while((c<'0'||c>'9')&&c!=-1)c=getchar();
    if(c==-1)exit(0);
    while(c>='0'&&c<='9'){
        k=k*10+c-'0';
        c=getchar();
    }
    return k;
}
int main(){
    int i,x,y;
    while(scanf("%d%d",&n,&m)!=EOF){
        for(i=1;i<=n;i++){
            a[i].clear();
            scanf("%d",&v[i]);
        }
        for(i=1;iscanf("%d%d",&x,&y);
            a[x].push_back(y);
            a[y].push_back(x);
        }
        memset(f0,0,sizeof(f0));
        memset(f1,0,sizeof(f1));
        memset(p,0,sizeof(p));
        dfs(1);
        printf("%d\n",f1[1][m]);
    }
    return 0;
}

（代码有点丑，一对堆文件（stdc++.h找的））