题意:给你一颗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);
}
}
}