题目大意:用m个机器人去遍历有n个节点的有根树,边权代表一个机器人通过这条边的代价,求最小代价。
题目分析:定义状态dp(root,k)表示最终遍历完成后以root为根节点的子树中有k个机器人时产生的总代价。则状态转移方程为:
dp(root,k)=min(dp(root,k),dp(son,j)+dp(root,k-j)+j*w(root,son)) j>0
要注意,当j为0的时候表示遍历完son这个子树后所有的机器人都回到root。可以证明,如果让遍历son的所有的机器人都回到root,那么遍历son的机器人越多,产生的代价就越高。
代码如下:
# include<iostream>
# include<cstdio>
# include<cstring>
# include<algorithm>
using namespace std;
# define LL long long
const int N=10005;
struct Edge
{
int to,w,nxt;
};
Edge e[N<<1];
int head[N];
int cnt,n,s,m;
int dp[N][12];
void add(int u,int v,int w)
{
e[cnt].to=v;
e[cnt].w=w;
e[cnt].nxt=head[u];
head[u]=cnt++;
}
void init()
{
int a,b,w;
cnt=0;
memset(dp,0,sizeof(dp));
memset(head,-1,sizeof(head));
for(int i=1;i<n;++i){
scanf("%d%d%d",&a,&b,&w);
add(a,b,w);
add(b,a,w);
}
}
void dfs(int u,int fa)
{
for(int i=head[u];i!=-1;i=e[i].nxt){
int v=e[i].to;
if(v==fa) continue;
dfs(v,u);
for(int j=m;j>=0;--j){
dp[u][j]+=dp[v][0]+2*e[i].w;
for(int k=1;k<=j;++k)
dp[u][j]=min(dp[u][j],dp[u][j-k]+dp[v][k]+k*e[i].w);
}
}
}
void solve()
{
dfs(s,-1);
printf("%d\n",dp[s][m]);
}
int main()
{
while(~scanf("%d%d%d",&n,&s,&m))
{
init();
solve();
}
return 0;
}