二*蘋果樹[樹型DP]
阿新 • • 發佈:2018-12-31
有一棵蘋果樹,如果樹枝有分叉,一定是分2叉(就是說沒有隻有1個兒子的結點)
這棵樹共有N個結點(葉子點或者樹枝分叉點),編號為1-N,樹根編號一定是1。
我們用一根樹枝兩端連線的結點的編號來描述一根樹枝的位置。下面是一顆有4個樹枝的樹
2 5
\ /
3 4
\ /
1
現在這顆樹枝條太多了,需要剪枝。但是一些樹枝上長有蘋果。
給定需要保留的樹枝數量,求出最多能留住多少蘋果。
輸入
第1行2個數,N和Q(1<=Q<= N,1
思路:樹形DP
設f[i,j]表示第i節點保留j子節點(包括本身和父親節點(只有根節點沒有父親節點))的最大蘋果數量。
轉移方程:f[i,j]=MAX{f[son,k]+f[son’,j-k-1]+sum;}
其中son和son’是兩個子節點,sum是父親節點到本節點的蘋果數量。
程式碼
var n,q,i,u,v,d:longint;
a,g:array[0..100,0..100]of longint;
f:array[0..100,0..100] of longint;
bz:array[0..100] of boolean;
procedure dg(x,last:longint);
var i,j,k,l,r,sum:longint;
begin
bz[x]:=false ;
l:=0;
r:=0;
for i:=1 to a[x,0] do
if bz[a[x,i]]then
begin
if l>0 then
begin
r:=a[x,i];
break;
end;
l:=a[x,i];
end;
if l>0 then
begin
if last=0 then sum:=0 else sum:=g[last,x];
dg(l,x);dg(r,x);
for j:=1 to q+1 do
for k:=0 to j-1 do if f[x,j]<f[l,k]+f[r,j-k-1]+sum then f[x,j]:=f[l,k]+f[r,j-k-1]+sum;
end
else f[x,1]:=g[last,x];
end;
begin
assign(input,'apple.in');reset(input);
assign(output,'apple.out');rewrite(output);
readln(n,q);
for i:=1 to n-1 do
begin
readln(u,v,d);
inc(a[u,0]);
a[u,a[u,0]]:=v;
g[u,v]:=d;
inc(a[v,0]);
a[v,a[v,0]]:=u;
g[v,u]:=d;
end;
fillchar(bz,sizeof(bz),1);
dg(1,0);
writeln(f[1,q+1]);
end.