[HNOI2004]宠物收养所 SBT
转载请注明出处，谢谢 http://blog.csdn.net/ACM_cxlove?viewmode=contents by---cxlove
继续跪，在找前驱和后继的时候出错了，debug出翔了，最后下载了测试数据对拍。
在找前驱和后继的时候，可能出现前驱和后继不存在，应当返回inf才对。
每于一棵树，要么为空，要么全为宠物或者全为人，加一个标记，表示树内现在的种类，对于一个新进来的，判断是否一致，一致则插入，否则找出一个相邻的删掉。
#include<iostream>#include<cstring>#include<cstdio>#include<algorithm>#define N 100005#define MOD 1000000#define inf 1<<29using namespace std;struct SBT{//左子树指针，右子树指针，大小，键值int left,right,size,key;void Init(){left=right=key=0;size=1;}}T[N];int root,tot; //根的位置以及节点个数//左旋转处理void Left_rot(int &x){ int k=T[x].right;T[x].right=T[k].left;T[k].left=x;T[k].size=T[x].size;T[x].size=T[T[x].left].size+T[T[x].right].size+1;x=k;}//右旋转处理void Right_rot(int &x){int k=T[x].left;T[x].left=T[k].right;T[k].right=x;T[k].size=T[x].size;T[x].size=T[T[x].left].size+T[T[x].right].size+1;x=k;}//调整处理void Maintain(int &r,bool flag){if(flag){ //更新右子树if(T[T[T[r].right].right].size>T[T[r].left].size)Left_rot(r);else if(T[T[T[r].right].left].size>T[T[r].left].size){Right_rot(T[r].right);Left_rot(r);}elsereturn;} else{ //更新在左子树if(T[T[T[r].left].left].size>T[T[r].right].size)Right_rot(r);else if(T[T[T[r].left].right].size>T[T[r].right].size){Left_rot(T[r].left);Right_rot(r);}else return;}//更新子树，然后再更新根，直到平衡为止Maintain(T[r].left,false);Maintain(T[r].right,true);Maintain(r,false);Maintain(r,true);}//插入新节点void Insert(int &r,int k){if(r==0){r=++tot;T[r].Init();T[r].key=k;}else{T[r].size++;if(k<T[r].key)Insert(T[r].left,k);elseInsert(T[r].right,k);//插入后要调整，保证平衡Maintain(r,k>=T[r].key);}}//删除结点，利用的是前驱替换int Remove(int &r,int k){int d_key;if(!r)return 0;T[r].size--;//前者说明就是要删的节点，后两者说明不存在此节点if(T[r].key==k||(T[r].left==0&&k<T[r].key)||(T[r].right==0&&k>T[r].key)){d_key=T[r].key;if(T[r].left&&T[r].right)T[r].key=Remove(T[r].left,k+1);elser=T[r].left+T[r].right;return d_key;}else return Remove(k<T[r].key?T[r].left:T[r].right,k);}void Delete(int &r,int delay,int min_val){if(!r) return;if(T[r].key+delay<min_val) {r=T[r].right;Delete(r,delay,min_val);}else{Delete(T[r].left,delay,min_val);T[r].size=T[T[r].right].size+T[T[r].left].size+1;}}//取得最大值，即一直遍历到最右的结点int Get_Max(int &r){while(T[r].right)r=T[r].right;return r;}//取得最小值，即一直遍历到最左的结点int Get_Min(int &r){while(T[r].left)r=T[r].left;return r;}//获得前驱int Get_Pre(int &r,int y,int k){if(r==0){if(y==0)return inf;return T[y].key;}if(k>=T[r].key)return Get_Pre(T[r].right,r,k);elsereturn Get_Pre(T[r].left,y,k);}//获得后继int Get_Next(int &r,int y,int k){if(r==0) {if(y==0)return inf;return T[y].key;}if(k<T[r].key)return Get_Next(T[r].left,r,k);elsereturn Get_Next(T[r].right,y,k);}//取得第K小的数，注：暂不能解决有重复数的int Get_Min_Kth(int &r,int k){int t=T[T[r].left].size+1;if(t==k) return T[r].key;if(t<k) return Get_Min_Kth(T[r].right,k-r);else return Get_Min_Kth(T[r].left,k);}//取得第K大的数int Get_Max_Kth(int &r,int k){int t=T[T[r].right].size+1;if(t==k) return T[r].key;if(t<k) return Get_Max_Kth(T[r].left,k-t);else return Get_Max_Kth(T[r].right,k);}//获得结点的名次int Get_Rank(int &r,int k){if(k<T[r].key) return Get_Rank(T[r].left,k);else if(k>T[r].key)return Get_Rank(T[r].right,k)+T[T[r].left].size+1;elsereturn T[T[r].left].size+1;}//查找树中是否存在元素int Find(int &r,int k){if(!r) return 0;if(T[r].key==k) return 1;if(k<T[r].key) return Find(T[r].left,k);else return Find(T[r].right,k);}//排序void Inorder(int &r){if(r==0) return;Inorder(T[r].left);printf("%d\n",T[r].key);Inorder(T[r].right);}int main(){int n,kind,k;int TreeKind;freopen("INPUT.001","r",stdin);while(scanf("%d",&n)!=EOF){tot=root=0;int ans=0,sum=0;while(n--){scanf("%d%d",&kind,&k);if(T[root].size==0||TreeKind==kind){Insert(root,k);TreeKind=kind;}else{if(Find(root,k)){Remove(root,k);continue;}int a=Get_Pre(root,0,k);int b=Get_Next(root,0,k);if(abs(a-k)<=abs(b-k)){ans=(ans+abs(a-k))%MOD;Remove(root,a);}else{ans=(ans+abs(b-k))%MOD;Remove(root,b);}}}printf("%d\n",ans);}return 0;}