树形dp--清洁机器人 nkoj3695

来源:互联网 发布:打车软件数据库设计 编辑:程序博客网 时间:2024/03/29 01:50
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;}


0 0
原创粉丝点击