[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<<29
using 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;
}

[HNOI2004]宠物收养所 SBT相关推荐

cogs62 [HNOI2004] 宠物收养所
cogs62 [HNOI2004] 宠物收养所啦啦啦啦不维护区间的平衡树题都是树状数组+二分练手题! 不会的参考我的普通平衡树的多种神奇解法之BIT+二分答案 // It is made by X ...
BZOJ 1208: [HNOI2004]宠物收养所
1208: [HNOI2004]宠物收养所 Time Limit: 10 Sec Memory Limit: 162 MB Submit: 7684 Solved: 3042 [Submit][S ...
1208: [HNOI2004]宠物收养所
1208: [HNOI2004]宠物收养所 1.用Treap树写: 只需要三个操作,插入,删除,查找(同时找出其前继后继): View Code /************************** ...
BZOJ 1208: [HNOI2004]宠物收养所 (Treap)
BZOJ 1208: [HNOI2004]宠物收养所题目概述: 有一家宠物收养所,提供两种服务:收养主人遗弃的宠物和让新主人领养宠物. 宠物收养所中总是会有两种情况发生:遗弃宠物过多和领养宠物人过多 ...
BZOJ1208[HNOI2004]宠物收养场——treap
凡凡开了一间宠物收养场.收养场提供两种服务:收养被主人遗弃的宠物和让新的主人领养这些宠物. 每个领养者都希望领养到自己满意的宠物,凡凡根据领养者的要求通过他自己发明的一个特殊的公式,得出该领养者希望领 ...
洛谷P2286 [HNOI2004]宠物收养所 [STL，平衡树]
题目传送门宠物收养所题目描述凡凡开了一间宠物收养场.收养场提供两种服务:收养被主人遗弃的宠物和让新的主人领养这些宠物. 每个领养者都希望领养到自己满意的宠物,凡凡根据领养者的要求通过他自己发明的 ...
bzoj1208: [HNOI2004]宠物收养所
Time Limit: 10 Sec Memory Limit: 162 MB Submit: 6182 Solved: 2396 [Submit][Status][Discuss] Descri ...
B1208 [HNOI2004]宠物收养所平衡树||set （滑稽）
这个题是一道splay裸题,但是我不太会写,所以用set直接水过去!!!哈哈哈哈,美滋滋. set总结: set是一个集合,然后里面没用重复的元素.里面有一些函数: begin() ,返回se ...
洛谷P2286 [HNOI2004]宠物收养场
题目描述凡凡开了一间宠物收养场.收养场提供两种服务:收养被主人遗弃的宠物和让新的主人领养这些宠物. 每个领养者都希望领养到自己满意的宠物,凡凡根据领养者的要求通过他自己发明的一个特殊的公式,得出该领 ...

[HNOI2004]宠物收养所 SBT

[HNOI2004]宠物收养所 SBT相关推荐

最新文章

热门文章