POJ2486Apple tree题解

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

#include <cctype>
#include <cerrno>
#include <cfloat>
#include <ciso646>
#include <climits>
#include <clocale>
#include <cmath>
#include <csetjmp>
#include <csignal>
#include <cstdarg>
#include <cstddef>
#include <cstdio>
#include <cstdlib>
#include <cstring>
#include <ctime>
#include <algorithm>
#include <bitset>
#include <complex>
#include <deque>
#include <exception>
#include <fstream>
#include <functional>
#include <iomanip>
#include <ios>
#include <iosfwd>
#include <iostream>
#include <istream>
#include <iterator>
#include <limits>
#include <list>
#include <locale>
#include <map>
#include <memory>
#include <new>
#include <numeric>
#include <ostream>
#include <queue>
#include <set>
#include <sstream>
#include <stack>
#include <stdexcept>
#include <streambuf>
#include <string>
#include <typeinfo>
#include <utility>
#include <valarray>
#include <vector>
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;i<a[x].size();i++){
        d=a[x][i];
        if(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;i<n;i++){
            scanf("%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找的））