apple tree POJ2486 树形dp

题目链接

题目大意：给你一颗树，树上每个点有一个点权，让你从1号点走k步，问可以得到的最大点权和是多少？

思路：这道题既然是个树结构，我们很容易想到是树形dp

设计状态：
由于我们走到一个点有可能是要回到这个点走到另一颗子树上去，也有可能不回来。所以要设两个状态：
1.$dp[0][x][s]$表示以x为根节点的子树走s步回到x点可以得到的最大点权和
2.$dp[1][x][s]$表示以x为根节点的子树走s步不回到x点可以得到的最大点权和

设计转移：
这里由于我们并不关心我们这个最大值是怎么走出来的，那么我们就可以用背包的思想，对于一颗子树，以根节点的儿子为根的子树所获得的最大点权和作为价值，枚举出所有给它的步数（步数作为体积）来进行转移，并且用已经转移的所有子树看作一个整体与新考虑的子树合并

dp转移方程：k为x的一个儿子
1.走到x点，回到x点：$dp[0][x][j+2]=max(dp[0][x][j+2],dp[0][k][t]+dp[0][x][j-t]$
这里之所以要+2，是因为从根节点到儿子再回来要用两步
2.不回到x点，但是回到k点之后要回到x点以向剩下的子树走：$dp[1][x][j+2]=max(dp[1][x][j+2],dp[1][x]j-t]+dp[0][k][t]$
3.不回到x点，也不会到k点，一直走下去：$dp[1][x][j+1]=max(dp[1][x][j+1],dp[0][x][j-t]+dp[1][k][t]$
我们是遍历完别的要回到根节点的（也就是2情况）之后再遍历这个子树，所以要加的是$dp[0][x][j-t]$而不是$dp[1][x][j-t]$

下附AC代码：

#include<iostream>
#include<vector>
#include<cstdio>
#include<cstring>
using namespace std;
int read(){
	char s;
	int x=0,f=1;
	s=getchar();
	while(s<'0'||s>'9'){
		if(s=='-')f=-1;
		s=getchar();
	}
	while(s>='0'&&s<='9'){
		x*=10;
		x+=s-'0';
		s=getchar();
	}
	return x*f;
}
int n,m;
int c[110];
vector<int>v[110];
int dp[2][110][210];//01分别代表是否回到该节点，并以该节点为根的子树用k步可以得到的最大价值 
void dfs(int x,int fa){
	for(int i=0;i<=m;i++){
		dp[0][x][i]=dp[1][x][i]=c[x];
	}
	for(int i=0;i<v[x].size();i++){
		int k=v[x][i];
		if(k==fa)continue;
		dfs(k,x);
		for(int j=m;j>=0;j--){
			for(int t=0;t<=j;t++){
				dp[0][x][j+2]=max(dp[0][x][j+2],//从子树回来也回到根节点 
				dp[0][x][j-t]+dp[0][k][t]);
				dp[1][x][j+2]=max(dp[1][x][j+2],//从子树回来，不回到根节点 
				dp[0][k][t]+dp[1][x][j-t]);
				dp[1][x][j+1]=max(dp[1][x][j+1],//不从子树回来，一路走到底 
				dp[1][k][t]+dp[0][x][j-t]);
			}
		}
	}
}
int main(){
	while(cin>>n){
		m=read();
		memset(dp,0,sizeof(dp));
		for(int i=1;i<=n;i++){
			c[i]=read();
			v[i].clear();
		}
		for(int i=1;i<n;i++){
			int a,b;
			a=read();
			b=read();
			v[a].push_back(b);
			v[b].push_back(a);
		}
		dfs(1,0);
		cout<<max(dp[0][1][m],dp[1][1][m])<<endl;
	}
}