题意/Description:
有一棵苹果树,如果树枝有分叉,一定是分2叉(就是说没有只有1个儿子的结点)
这棵树共有N个结点(叶子点或者树枝分叉点),编号为1-N,树根编号一定是1。
我们用一根树枝两端连接的结点的编号来描述一根树枝的位置。下面是一颗有4个树枝的树
2 5
\ /
3 4
\ /
1
现在这颗树枝条太多了,需要剪枝。但是一些树枝上长有苹果。给定需要保留的树枝数量,求出最多能留住多少苹果。
读入/Input:
第1行2个数,N和Q。 N表示在树枚举的点数。Q表示应保留的分支数量。
下来N - 1行中包含分支的描述。每个描述包括三个整数用空格分开的。它们的前两个由它的结束点定义分支。第三个数字上定义这个分支苹果的数量。你可以假设没有分支包含超过30000苹果。
输出/Output:
一个数,最多能留住的苹果的数量。
题解/solution:
这个题目都讲到了二叉了,很容易的想到树型DP。不讲废话,进入话题。
如果只有一个容量,取左右两边最大值。
f[t,x]:=max(g[t,l[t]],g[t,r[t]]);
如果大于一个容量,先设左右两边都被分配。
f[t,x]:=max(f[t,x],f[l[t],i]+f[r[t],x-2-i]);
然后只取左边或只取右边。
f[t,x]:=max(f[t,x],f[l[t],x-1]+g[t,l[t]]); 或 f[t,x]:=max(f[t,x],f[r[t],x-1]+g[t,r[t]]);
代码/Code:
<strong>var n,q,o:longint; a,g,f:array [0..101,0..101] of longint; v:array [0..101] of boolean; l,r,du:array [0..101] of longint; function max(t,k:longint):longint; begin if t>k then exit(t); exit(k); end; procedure tree(x:longint); var i:longint; begin if x=0 then exit; v[x]:=true; for i:=1 to a[x,0] do if not v[a[x,i]] then if l[x]=0 then l[x]:=a[x,i] else r[x]:=a[x,i]; tree(l[x]); tree(r[x]); end; procedure main(t,x:longint); var i:longint; begin if (x=0) or (t=0) then begin f[t,x]:=0; exit; end; if f[t,x]>0 then exit; if x=1 then begin f[t,x]:=max(g[t,l[t]],g[t,r[t]]); exit; end; for i:=0 to x-2 do begin main(l[t],i); main(r[t],x-2-i); f[t,x]:=max(f[t,x],f[l[t],i]+f[r[t],x-2-i]); end; f[t,x]:=f[t,x]+g[t,l[t]]+g[t,r[t]]; main(l[t],x-1); f[t,x]:=max(f[t,x],f[l[t],x-1]+g[t,l[t]]); main(r[t],x-1); f[t,x]:=max(f[t,x],f[r[t],x-1]+g[t,r[t]]); end; procedure init; var i,x,y,z:longint; begin readln(n,q); for i:=1 to n-1 do begin readln(x,y,z); g[x,y]:=z; g[y,x]:=z; inc(a[x,0]); a[x,a[x,0]]:=y; inc(a[y,0]); a[y,a[y,0]]:=x; end; o:=1; for i:=1 to n do if a[i,0]=2 then begin o:=i; break; end; end; begin init; tree(o); main(o,q); write(f[o,q]); end. </strong>
时间: 2024-10-29 12:57:10