【问题描述】
传说中的九头龙是一种特别贪吃的动物。虽然名字叫“九头龙”,但这只是说它出生的时候有九个头,而在成长的过程中,它有时会长出很多的新头,头的总数会远大于九,当然也会有旧头因衰老而自己脱落。
有一天,有M个脑袋的九头龙看到一棵长有N个果子的果树,喜出望外,恨不得一口把它全部吃掉。可是必须照顾到每个头,因此它需要把N个果子分成M组,每组至少有一个果子,让每个头吃一组。
这M个脑袋中有一个最大,称为“大头”,是众头之首,它要吃掉恰好K个果子,而且K个果子中理所当然地应该包括唯一的一个最大的果子。果子由N-1根树枝连接起来,由于果树是一个整体,因此可以从任意一个果子出发沿着树枝“走到”任何一个其他的果子。
对于每段树枝,如果它所连接的两个果子需要由不同的头来吃掉,那么两个头会共同把树枝弄断而把果子分开;如果这两个果子是由同一个头来吃 掉,那么这个头会懒得把它弄断而直接把果子连同树枝一起吃掉。当然,吃树枝并不是很舒服的,因此每段树枝都有一个吃下去的“难受值”,而九头龙的难受值就 是所有头吃掉的树枝的“难受值”之和。
九头龙希望它的“难受值”尽量小,你能帮它算算吗?
例如图1所示的例子中,果树包含8个果子,7段树枝,各段树枝的“难受值”标记在了树枝的旁边。九头龙有两个脑袋,大头需要吃掉4个果子,其中必须包含最大的果子。即N=8,M=2,K=4:
大头吃4个果子,用实心点标识;
小头吃4个果子,用空心点标识;
九头龙的难受值为4,因为图中用细边标记的树枝被大头吃掉了。
图一描述了果树的形态,图二描述了最优策略。
【输入文件】
输入文件的第1行包含三个整数N (1<=N<=300),M (2<=M<=N),K (1<=K<=N)。 N个果子依次编号1,2,...,N,且最大的果子的编号总是1。第2行到第N行描述了果树的形态,每行包含三个整数a (1<=a<=N),b (1<=b<=N),c (0<=c<=105),表示存在一段难受值为c的树枝连接果子a和果子b。
【输出文件】
输出文件仅有一行,包含一个整数,表示在满足“大头”的要求的前提下,九头龙的难受值的最小值。如果无法满足要求,输出-1。
【样例输入】
8 2 4 1 2 20 1 3 4 1 4 13 2 5 10 2 6 12 3 7 15 3 8 5
【样例输出】
4时间限制:1 s 内存限制:128 MB题目大意:有一只M个头的九头龙,想吃掉有N个果实的树,要求每个小头至少吃1个果实,大头吃恰好K个果实,且树根上的果实也需由大头吃。对于一个果实,如果同一个头同时吃掉这个果实以及这个果实的父亲,就会受到一定的难受值,现在求吃掉这棵树的最小难受值。 首先判断无解,很明显仅当N<M+K-1时会无解。题目中头的数目2<=M<=N咋一看可能会被吓到,感觉无从下手,但仔细想一想M个脑袋是可以简化掉的,大致有两种情况:①M=2时 这时候大头和小头的难受值都需要考虑,但因为只有两个头还是可以解决的。②M>2时 这时候只需要考虑大头的难受值,因为小头完全可以避开吃掉树枝。然后为了方便转移,我们把可以把这棵多叉树转换成二叉树,这样问题就简单多了。到了这里这道题目就变成一道简单的树形DP了f[i][j][1/0] 表示以i为根的子树中,大头要吃j个果实,且其父结点也被大头吃了(1)或没被大头吃(0),的最小难受值转移方程详见程序,这里说个要注意的地方:对于一棵大头不吃任何果实的树,若M>2可以直接得出难受值为0,但是若M==2便还是要遍历到叶结点才结束。
1 #include<cstdio> 2 #include<cstring> 3 #include<algorithm> 4 #define R register 5 6 struct info{ 7 int to,pre,dis; 8 }edge[1007]; 9 10 bool pd[307][307]; 11 int n,m,k,size,last[307],head[307]; 12 int f[307][307][2],fa[307],l[307],r[307],up[307]; 13 14 void addline(int from,int to,int dis){ 15 edge[++size].pre=head[from];head[from]=size; 16 edge[size].to=to;edge[size].dis=dis; 17 edge[++size].pre=head[to];head[to]=size; 18 edge[size].to=from;edge[size].dis=dis; 19 } 20 21 void tsdp(int k,int num){ 22 if (pd[k][num]) return; 23 f[k][num][1]=f[k][num][0]=0x7ffffff; 24 if (!k||!num){ //处理0结点和大头不吃果实的情况 25 if (!num&&(m!=2||!k)) f[k][num][0]=f[k][num][1]=0; 26 if (m!=2||!k)return; 27 } 28 if (!l[k]&&!r[k]){ //处理叶结点 29 if (num==1) f[k][1][1]=up[k],f[k][1][0]=0; 30 if (!num) f[k][0][0]=up[k],f[k][0][1]=0; 31 } 32 for (R int i=0;i<=num;++i){ //状态转移 33 tsdp(l[k],i); 34 tsdp(r[k],num-i); 35 f[k][num][1]=std::min(f[k][num][1],f[l[k]][i][0]+f[r[k]][num-i][1]); 36 if (i) 37 f[k][num][1]=std::min(f[k][num][1],f[l[k]][i-1][1]+up[k]+f[r[k]][num-i][1]); 38 if (i) 39 f[k][num][0]=std::min(f[k][num][0],f[l[k]][i-1][1]+f[r[k]][num-i][0]); 40 if (m==2) 41 f[k][num][0]=std::min(f[k][num][0],f[l[k]][i][0]+up[k]+f[r[k]][num-i][0]); 42 else 43 f[k][num][0]=std::min(f[k][num][0],f[l[k]][i][0]+f[r[k]][num-i][0]); 44 } 45 pd[k][num]=true; 46 } 47 48 void DFS(int k){ 49 for (R int i=head[k];i;i=edge[i].pre) 50 if (edge[i].to!=fa[k]){ 51 fa[edge[i].to]=k; 52 DFS(edge[i].to); 53 up[edge[i].to]=edge[i].dis; 54 } 55 } 56 57 int main(){ 58 scanf("%d%d%d",&n,&m,&k); 59 for (R int i=1;i<n;++i){ 60 int a,b,c; 61 scanf("%d%d%d",&a,&b,&c); 62 addline(a,b,c); 63 } 64 DFS(1); 65 for (R int i=2;i<=n;++i){ //多叉转二叉 66 if (last[fa[i]]) r[last[fa[i]]]=i; 67 else l[fa[i]]=i; 68 last[fa[i]]=i; 69 } 70 tsdp(1,k); 71 if (n<k+m-1) printf("-1"); 72 else printf("%d",f[l[1]][k-1][1]); 73 }
PS:这个程序写得太丑了,又臭又长,但又不想重写,凑合着看一下吧TAT
评测地址:https://www.vijos.org/p/1523