sbt-size balanced tree学习笔记

sbt的详尽资料：http://www.nocow.cn/index.php/Size_Balanced_Tree

个人理解：

先贴自己写的（虽然和cqf大牛的程序如出一辙..原因是太佩服cqf大牛的写法了.）

 const maxn=1000000; var key,left,right,s:array[1..maxn] of longint; n,tt,x:longint; procedure main; var h,t,i:longint; procedure right_rotate(var t:longint); begin h:=left[t]; left[t]:=right[h]; right[h]:=t; s[h]:=s[t]; s[t]:=s[left[t]]+s[right[t]]+1;//Attention! t:=h; end; procedure left_rotate(var t:longint); begin h:=right[t]; right[t]:=left[h]; left[h]:=t; s[h]:=s[t]; s[t]:=s[left[t]]+s[right[t]]+1;//Attention! t:=h; end; procedure maintain(var t:longint;flag:boolean); begin if not flag then if s[left[left[t]]]>s[right[t]] then right_rotate(t) else if s[right[left[t]]]>s[right[t]] then begin left_rotate(left[t]); right_rotate(t); end else exit else if s[right[right[t]]]>s[left[t]] then left_rotate(t) else if s[left[right[t]]]>s[left[t]] then begin right_rotate(right[t]); left_rotate(t); end else exit; maintain(left[t],false); maintain(right[t],true); maintain(t,true); maintain(t,false); end; procedure insert(var t,v:longint); begin if t=0 then begin inc(tt); t:=tt; key[t]:=v; s[t]:=1; left[t]:=0; right[t]:=0; exit; end; if v<key[t] then insert(left[t],v) else insert(right[t],v); maintain(t,v>=key[t]); end; procedure print(t:longint); begin if t=0 then exit; print(left[t]); write(key[t],' '); print(right[t]); end; function succ(var t:longint;v:longint):longint; begin if t=0 then exit(v); if v>=key[t] then succ:=succ(right[t],v) else begin succ:=succ(left[t],v); if succ=v then succ:=key[t]; end; end; begin readln(n); tt:=0; t:=0; for i:=1 to n do begin read(x); insert(t,x); end; print(t); writeln(succ(t,1)); end; begin main; end. 

 这个代码只实现了插入、求前驱、排序输出三个功能，但核心部分已经写好了----maintain。其他几个功能的实现，可以参考cqf大牛的官方程序。

attention部分：一开始没有马上理解这个语句，仔细看了下后才理解。这个语句之前，t的左右儿子已经发生了改变，所以顺理成章地，s[t]等于它的左右儿子大小之和加1.佩服cqf大牛写程序时清晰的逻辑！

对insert(var t,v:longint)的理解：这里t前面有个var，使得插入新节点的过程中，该节点的父亲“自动”与它相连，使得代码精炼高效；每次递归完之后，t的值都会被赋值为树根（在maintain中），相当巧妙。

对maintain的理解：对树进行调整的过程，是整个程序中最关键的过程之一。flag用于减少无用计算，属于常数优化。需要优化的情况可以分为两大类，共四种情况，分别用对应maintain过程中的四个if中的旋转操作再递归作用就行了。最后为什么不用maintain(left[t],true)和maintain(right[t],false)，cqf大牛的论文里有提及。自己画图看看就明白了。