【TreeDP】没有上司的晚会 NDK1290 题解

program NDK1290; const MaxN=6010; var root,n:longint; tot:array [-1..MaxN,-1..150] of longint;//邻接表，存储每个上司的下属 f:array [-1..MaxN,0..1] of longint;//f[x,0]表示第x号人不去，f[x,1]表示第x号人去 b:array [-1..MaxN] of boolean;//用来找根 a:array [-1..MaxN] of longint; procedure init; var i,l,k:longint; begin fillchar(tot,sizeof(tot),0); fillchar(f,sizeof(f),255);//当一个人的f[x,y]的值为0时，也可以表示这个人不去，因此初始化为-1，避免情况的混乱 fillchar(b,sizeof(b),false); readln(n); for i:=1 to n do readln(a[i]); readln(l,k); while (l<>0) and (k<>0) do begin inc(tot[k,0]); tot[k,tot[k,0]]:=l; b[l]:=true;//打标记 readln(l,k); end; for i:=1 to n do begin if not b[i] then begin root:=i;//根没在下属中出现过，那么找一个没在下属中出现过的i就是根 break; end; end; end; function max(x,y:longint):longint; begin if x>y then exit(x) else exit(y); end; function tree_dp(x,y:longint):longint; var i:longint; begin if f[x,y]>=0 then exit(f[x,y]);//记忆化优化时间复杂度，大于0表示已经有值，等于0表示第x个人不去 if tot[x,0]=0 then//tot[x,0]=0表示第x号人没有下属 begin f[x,y]:=a[x]*y;//若y=1，那么f[x,y]=a[x]表示x号人去了；若y=0，那么f[x,y]=0表示x号人没去 exit(f[x,y]); end; if y=0 then begin f[x,y]:=0;//x号人不去，不去所得的值初始为0 for i:=1 to tot[x,0] do inc(f[x,y],max(tree_dp(tot[x,i],0),tree_dp(tot[x,i],1)));//x号人不去，下属可以去也可以不去，递归找一个最大值 end else begin f[x,y]:=a[x];//x号人去了，所得的值初始为x号人的权值 for i:=1 to tot[x,0] do inc(f[x,y],tree_dp(tot[x,i],0));//x号人去了，下属一定不去 end; exit(f[x,y]);//之前是特殊情况的输出，这个是一般情况的输出 end; begin assign(input,'NDK1290.in'); reset(input); assign(output,'NDK1290.out'); rewrite(output); init; writeln(max(tree_dp(root,0),tree_dp(root,1)));//判断root是去取得的权值大还是不去取得的权值大 close(input); close(output); end.