1. 程式人生 > >二*蘋果樹[樹型DP]

二*蘋果樹[樹型DP]

有一棵蘋果樹,如果樹枝有分叉,一定是分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.