UVALive4015(树形dp)

#include <vector>
#include <stdio.h>
#include <stdlib.h>
#include <string.h>
#include <iostream>
#include <algorithm>
using namespace std;
const int INF=0x3f3f3f3f;
struct node{
int to,cost;
};
vector<node> G[505];
int ans[505],vis[505],sum[505],dp[505][505][2];
void dfs(int s){
int i,j,k,tmp;
vis[s]=1;
for(i=0;i<G[s].size();i++){
tmp=G[s][i].to;
if(vis[tmp])
continue;
dfs(tmp);
sum[s]+=sum[tmp];
for(j=sum[s];j>1;j--){
for(k=min(sum[tmp],j-1);k>=0;k--){
dp[s][j][0]=min(dp[s][j][0],dp[s][j-k][0]+dp[tmp][k][0]+2*G[s][i].cost);
dp[s][j][1]=min(dp[s][j][1],min(dp[s][j-k][1]+dp[tmp][k][0]+2*G[s][i].cost,dp[s][j-k][0]+dp[tmp][k][1]+G[s][i].cost));
}                                   //转移的时候一定要想清楚
}
}
}                                               //dp[i][j][0]表示走到i点，经过j个点后回到i点
int main(){                                     //dp[i][j][0]表示走到i点，经过j个点后不回到i点
int n,i,j,x,y,z,q,cas,tmp;
cas=1;
while(scanf("%d",&n)!=EOF&&n){
for(i=0;i<n;i++)
G[i].clear();
for(i=1;i<n;i++){
scanf("%d%d%d",&x,&y,&z);
G[y].push_back((node){x,z});
G[x].push_back((node){y,z});
}
memset(dp,INF,sizeof(dp));
for(i=0;i<=n;i++){
vis[i]=0,sum[i]=1;
dp[i][1][0]=dp[i][1][1]=0;
}
dfs(0);
for(i=1;i<=n;i++)                       //ans[i]为走i个点最近的路程
ans[i]=min(dp[0][i][0],dp[0][i][1]);
scanf("%d",&q);
printf("Case %d:\n",cas++);
while(q--){
scanf("%d",&x);
tmp=1;
for(i=1;i<=n;i++){                  //也可以二分找
if(ans[i]<=x)
tmp=i;
else
break;
}
printf("%d\n",tmp);
}
}
return 0;
}