【APIO】风铃(mobiles)

 

 

 

---

 

二叉树，而且每个点的子节点不会只有一个（没有度为1的点，不是0就是2）。

从子节点开始向上进行，每次判断当前的节点可以调整或者非法（无法调整或者是深度相差超过1）。

非法的情况有

1.无法调整——两个子树都是不平衡的

2.两颗子树深度差超过了1。

 

记录以下几个量

d[i]:i结点的子树的最深深度

b[i]:i结点是否平衡 不平衡为true 平衡为false

t[i]:i结点的子结点的个数——>用来判断结点是否都是平的，仅仅靠b数组是无法由子结点来判断当前结点是否平衡的

 

我们需要注意，题目中说明子结点都比父节点的编号大，所以我们只要从n downto 1一个循环就可以了，用dfs的方法两个测试点由于递归过深造成了崩溃。

 

程序以下

var t,d,l,r:array[-1..100000] of longint; b:array[-1..100000] of boolean; n,ans:longint; procedure readin; var i:longint; begin readln(n); for i:=1 to n do readln(l[i],r[i]); end; function max(a,b:longint):longint; begin if a>b then exit(a); exit(b); end; procedure fail; begin writeln(-1); close(input); close(output); halt; end; procedure main; var x,lt,rt,ld,rd:longint; lb,rb:boolean; begin for x:=n downto 1 do begin if l[x]=-1 then begin lt:=1; ld:=1; lb:=false; end else begin lt:=t[l[x]]; ld:=d[l[x]]; lb:=b[l[x]]; end; if r[x]=-1 then begin rt:=1; rd:=1; rb:=false; end else begin rt:=t[r[x]]; rd:=d[r[x]]; rb:=b[r[x]]; end; if (l[x]<>-1) and (r[x]<>-1) and rb and lb then fail; if abs(ld-rd)>1 then fail; if lt=rt then b[x]:=false else b[x]:=true; if lt<rt then inc(ans); t[x]:=lt+rt; d[x]:=max(ld,rd)+1; end; end; begin assign(input,'.in'); reset(input); assign(output,'.out'); rewrite(output); readin; main; writeln(ans); close(input); close(output); end.

 

通过APIO官方的42个数据