uvalive 4015 树形背包dp

题意：给你一颗n个节点的树，求从根节点出发最多走x远，问最多能经过多少个节点。

分析:考虑dp方程dp[i][j][0]表示以i为根节点、经过j个节点、不回到i、最小要走多远,

dp[i][j][1]表示以i为根节点、经过j个节点、回到i、最小要走多远。

容易想到状态转移方程

dp[u][j][0] = min(dp[u][j][0],dp[u][j-k][1] + dp[v][k][0]+ c);
dp[u][j][0] = min(dp[u][j][0],dp[u][j-k][0] + dp[v][k][1]+ 2*c);
dp[u][j][1] = min(dp[u][j][1],dp[u][j-k][1] + dp[v][k][1]+ 2*c);

其中第二个方程表示，经过子树v并且返回u，然后走其他j-k个节点不反回。

细节详见代码。

#include<iostream>
#include<cstring>
#include<cstdio>
#include<algorithm>
#include<vector>
using namespace std;
const int maxn = 500;
const int INF = 0x3f3f3f3f;
typedef pair<int ,int> P;
int dp[maxn][maxn][2];
int sum[maxn]; 
vector<vector<P> > A(maxn);
void dfs(int u,int fa)
{
	sum[u] = 1;
	for(int i = 0; i < A[u].size();i++)
	{
		int v = A[u][i].first;
		if(v != fa)
		{
			dfs(v,u);
			sum[u] += sum[v];
		}
	}
	dp[u][1][0] = dp[u][1][1] = 0;
	for(int i = 0; i < A[u].size();i++)
	{
		int v = A[u][i].first,c = A[u][i].second ;
		if(v != fa)
		{
			for(int j = sum[u];j >= 1; j --)
			{
				for(int k = 1; k < j && k <= sum[v]; k ++)
				{
					dp[u][j][0] = min(dp[u][j][0],dp[u][j-k][1] + dp[v][k][0]+ c);
					dp[u][j][0] = min(dp[u][j][0],dp[u][j-k][0] + dp[v][k][1]+ 2*c);
					dp[u][j][1] = min(dp[u][j][1],dp[u][j-k][1] + dp[v][k][1]+ 2*c);
				}
			}
		}
	}
} 
int main(void) 
{
	int n,x,q,a,b,c,k = 0;
	while(~scanf("%d",&n) && n)
	{
		for(int i = 0; i < n;i++) A[i].clear();
		for(int i = 0; i < n - 1; i ++)
		{
			scanf("%d%d%d",&a,&b,&c);
			A[a].push_back(P(b,c));
			A[b].push_back(P(a,c));
		}
		memset(dp,0x3f,sizeof(dp));
		scanf("%d",&q);
		dfs(0,-1);
		printf("Case %d:\n",++k);  
		while(q--)
		{
			scanf("%d",&x);
			int L = 0,R = n;
			while(R > L)
			{
				int mid = L + (R - L + 1)/2;
				if(dp[0][mid][0] > x && dp[0][mid][1] > x)	R = mid - 1;
				else L = mid;
			}
			printf("%d\n",L);
		}
	}
}