【CodeVS1380】【Tyvj1052】没有上司的舞会

【Description】

　　Ural大学有N个职员，编号为1~N。他们有从属关系，也就是说他们的关系就像一棵以校长为根的树，父结点就是子结点的直接上司。每个职员有一个快乐指数。现在有个周年庆宴会，要求与会职员的快乐指数最大。但是，没有职员愿和直接上司一起与会。

---

【Input】

　　第一行一个整数N。(1<=N<=6000)
　　接下来N行，第i+1行表示i号职员的快乐指数Ri。(-128<=Ri<=127)
　　接下来N-1行，每行输入一对整数L,K。表示K是L的直接上司。
最后一行输入0,0。

---

【Output】

　　输出最大的快乐指数。

---

【Sample Input】

7 
1 
1 
1 
1 
1 
1 
1 
1 3 
2 3 
6 4 
7 4 
4 5 
3 5 
0 0

---

【Sample Output】

5

---

【Solution】

　　本题是简单的树形DP，对于每个节点只有选和不选两种可能。
　　
　　参考代码如下：

var n,i,l,k,root:longint;
    r:array[0..7000] of longint;
    f:array[0..7000,0..1] of longint;
    father:array[0..7000] of longint;
function max(x,y:longint):longint;
  begin if x>y then exit(x) else exit(y); end;
procedure search(x:longint); //树形DP
  var i:longint;
      flag:boolean;
  begin
    flag:=false;
    for i:=1 to n do begin
     if father[i]=x then begin
       flag:=true;
       search(i);
       f[x,0]:=f[x,0]+max(f[i,1],f[i,0]);
       f[x,1]:=f[x,1]+f[i,0];
     end;
    end;
    if not flag then begin
      f[x,0]:=0;
      f[x,1]:=r[x];
    end
    else f[x,1]:=f[x,1]+r[x];
  end;
begin
  readln(n);
  for i:=1 to n do readln(r[i]);
  for i:=1 to n-1 do begin
    readln(l,k);
    father[l]:=k;
  end;
  readln(l,k);
  for i:=1 to n do
    if father[i]=0 then begin root:=i; break; end;
  fillchar(f,sizeof(f),0);
  search(root);
  writeln(max(f[root,0],f[root,1]));
end.