平衡二叉数.docVIP

{treap示范代码： 标准的代码缩进风格，优美的算法实现。 经典标程，完全掌握后水平会提高很多 不改变bst的性质(在bst所有子树中均满足：左子树的所有节点=根=右子树的所有节点) 通过旋转操作，使根的hr最小(即所有的hr构成堆的关系) } var //l[i],r[i],v[i]:i号结点的左儿子、右儿子，关键值 //hr[i]:i号节点的优先值(treap所有子树中,根的hr必须是最小的) //s[i]:i号节点为根的子树节点总数 l,r,hr,s,v:array[0..2000000]of longint; n,root,m:longint; procedure init;//初始化 begin readln(n); m:=0; //randomize; //考试要求程序每次运行结果必须一致,慎用。确实要用：randomize 100; fillchar(s,sizeof(s),0); fillchar(l,sizeof(l),0); fillchar(r,sizeof(r),0); root:=0; end; //旋转是平衡二叉树的精髓，它不会改变bst的性质(左子树=根=右子树) //左旋使树的左子树深度+1,右子树深度-1 //右旋使树的右子树深度+1,左子树深度-1 procedure l_rotate(var x:longint);inline;//左旋以x为根的子树（注意var参数及意义） var y:longint; begin y:=r[x]; //保存x的右儿子到y中 r[x]:=l[y]; //将y的左儿子作为x的右儿子 l[y]:=x; //x作为y的左儿子 s[y]:=s[x]; //维护旋转后的子树大小 s[x]:=s[l[x]]+s[r[x]]+1; x:=y; //y为根 end; procedure r_rotate(var x:longint);inline;//右旋以x为根的子树 var y:longint; begin y:=l[x]; l[x]:=r[y]; r[y]:=x; s[y]:=s[x]; s[x]:=s[l[x]]+s[r[x]]+1; x:=y; end; //插入（递归，if key=root,则插入到左子树，否则到右子树，直到尽头再新建节点） procedure insert(var k,key:longint);inline; begin if k=0 then//已到尽头，新建节点并写入key及随机值hr begin inc(m); v[m]:=key; s[m]:=1; hr[m]:=random(maxlongint); k:=m;//修改k,使父节点指向当前节点(修改前从父节点指向0) exit; end; inc(s[k]); if key=v[k] then//若key=根则插入到左子树，否则到右子树 begin insert(l[k],key);//若l[k]=0,到下层则新建节点并修改l[k]=m if hr[l[k]]hr[k] then //旋转 r_rotate(k); exit; end; if keyv[k] then begin insert(r[k],key); if hr[r[k]]hr[k] then l_rotate(k); exit; end; end; (删除:在k号节点为根的子树中删除key 基本方法：由于是静态结构，为了提高效率，并没真正删除 若找到则删除，若没找到，则删除查找尽头的节点 主程序中需判断返回值，若不等于key,重新插入key即可 找到后的处理： 若为叶节点，直接删除，否则，将要删除的节点左子树的最右节点（思考：为什么？）代替它 ) function delete(var k:longint;key:longint):longint;inline; begin dec(s[k]);//维护节点总数 //如果找到，或已到尽头 if (key=v[k])or(l[k]=0)and(key=v[k])or(r[k]=0)and(keyv[k]) then begin delete:=v[k];//返回要删除的节点（不一定=key） if (l[k]=0)or(r[k]=0) then //