题目地址: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找的))