时间限制: 1 s
空间限制: 128000 KB
题目等级 : 钻石 Diamond
题目描述 Description
Ural大学有N个职员,编号为1~N。他们有从属关系,也就是说他们的关系就像一棵以校长为根的树,父结点就是子结点的直接上司。每个职员有一个快乐指数。现在有个周年庆宴会,要求与会职员的快乐指数最大。但是,没有职员愿和直接上司一起与会。
输入描述 Input Description
第一行一个整数N。(1<=N<=6000) 接下来N行,第i+1行表示i号职员的快乐指数Ri。(-128<=Ri<=127) 接下来N-1行,每行输入一对整数L,K。表示K是L的直接上司。 最后一行输入0,0。
输出描述 Output Description
输出最大的快乐指数。
样例输入 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
数据范围及提示 Data Size & Hint
各个测试点1s
题解:额。。最近快要省选了,感觉自己太弱,来发水题增加点信心(HansBug:其实是好久没打树形DP了有木有)。这题别的没啥,是一棵树,不是森林,而且有唯一的根节点,这就不难办了(假如是个根节点不唯一的有向图的话还得拓扑排序,唉那样子真心神烦。。。唉这年头写个水题还得想这么多,我真心太弱了TT)
1 type
2 point=^node;
3 node=record
4 g:longint;
5 next:point;
6 end;
7 vec=record
8 a0,a1:longint;
9 end;
10 var
11 i,j,k,l,m,n:longint;
12 d1,d2:vec;
13 a:array[0..10000] of point;
14 b,c:array[0..10000] of longint;
15 function max(x,y:longint):longint;inline;
16 begin
17 if x>y then max:=x else max:=y;
18 end;
19 function min(x,y:longint):longint;inline;
20 begin
21 if x<y then min:=x else min:=y;
22 end;
23 function dp(x:longint):vec;
24 var
25 d1,d2:vec;
26 p:point;
27 begin
28 d1.a0:=0;d1.a1:=b[x];
29 p:=a[x];
30 while p<>nil do
31 begin
32 d2:=dp(p^.g);
33 d1.a0:=d1.a0+max(d2.a1,d2.a0);
34 d1.a1:=d1.a1+d2.a0;
35 p:=p^.next;
36 end;
37 exit(d1);
38 end;
39 procedure add(x,y:longint);inline;
40 var p:point;
41 begin
42 new(p);p^.g:=y;
43 p^.next:=a[x];a[x]:=p;
44 end;
45 begin
46 readln(n);
47 for i:=1 to n do readln(b[i]);
48 for i:=1 to n do a[i]:=nil;
49 fillchar(c,sizeof(c),0);
50 for i:=1 to n-1 do
51 begin
52 readln(j,k);
53 add(k,j);c[j]:=1;
54 end;
55 m:=-1;
56 for i:=1 to n do if c[i]=0 then m:=i;
57 d1:=dp(m);
58 writeln(max(d1.a0,d1.a1));
59 end.
60