树形dp--清洁机器人 nkoj3695

P3695清洁机器人

时间限制 : - MS   空间限制 : 65536 KB  

评测说明 : 时限1000ms

问题描述

NK中学有n间教室（编号1到n），通过n-1条双向道路相连，每条道路的长度可能不同。现在有k台清洁机器人位于s号教室，现在要安排它们去清洁所有教室。
机器人靠燃油驱动，一台机器人清洁一个教室的耗油1升。在道路上行走时，每单位距离耗油1L。
我们希望完成清洁作业消耗的总油量尽可能少。请你计算出这个总油量。
作业结束时，机器人可以停留在任何位置。


输入格式

第一行，三个整数n,s,k
接下来n-1行，每行三个整数a,b,c，表示教室a和教室b之间有条长度为c的道路相连。


输出格式

一行，一个整数，表示所求答案。


样例输入 1

3 1 1
1 2 1
1 3 1


样例输出 1

6


样例输入 2

3 1 2
1 2 1
1 3 1


样例输出 2

5


提示

样例1说明，机器人的活动线路为1->2->1->3,路上耗费3L燃油，清洁教室耗费3L


1<=n<=10000, 1<=s<=N, 1<=k<=10

1<=c<=1000000.

#include<cstdio>
#include<iostream>
#include<algorithm>
#include<cstring>
using namespace std;
struct line{
	long long from,to,len;
};
line edge[1000005];
long long last[100005],_next[1000005];
long long f[100005][15];
long long n,s,k,m=0;
void add_edge(long long x,long long y,long long len){
	m++;
	_next[m]=last[x];
	last[x]=m;
	edge[m].from=x;
	edge[m].to=y;
	edge[m].len=len;
}
void dp(long long u,long long fa){
	long long i,j,h,v,t;
	for(h=last[u];h;h=_next[h]){
		v=edge[h].to;
		if(v==fa)continue;
		dp(v,u);
		for(j=k;j>=0;j--){
			f[u][j]+=f[v][0]+2*edge[h].len;
			for(t=0;t<=j;t++){
				f[u][j]=min(f[u][j],f[v][t]+f[u][j-t]+t*edge[h].len);
			}
		}
	}
}
int main(){
	long long i,j;
	cin>>n>>s>>k;
	for(i=1;i<n;i++){
		long long a,b,c;
		scanf("%I64d%I64d%I64d",&a,&b,&c);
		add_edge(a,b,c);
		add_edge(b,a,c);
	}
	dp(s,0);
	cout<<f[s][k]+n;
}