首页 > 代码库 > BZOJ 3196 二逼平衡树 树套树(线段树套Treap)
BZOJ 3196 二逼平衡树 树套树(线段树套Treap)
题目大意:
写一种数据结构,他可以:
1.查询k在区间内的排名。
2.查询区间内排名为k的值
3.修改某一个值。
4.求k在区间内的前驱。
5.求k在区间内的后继。
思路:本来以为有什么只有神犇才知道的神一般的数据结构来维护它,问了别人之后,发现只是树套树。据说怎么套都行。我见识鄙陋,就只能线段树套Treap了。
这也是第一次写树套树,还1A了,有点开心。
写树套树,一定要确定自己对这两个树及其熟练,加上少量精细的思考,就可以完成树套树。(我只是弱渣,求神犇别D)
具体实现:第一层是线段树,第二层是Treap。由于每个线段树的节点代表实际的一段区间,这样就可以在每个线段树的节点下面接一个Treap,维护这段区间,并可以在O(logn)的时间之内求出这段区间的各种信息。
考虑第一种操作:考虑线段树的工作原理,对于整段区间,就直接在这个区间的Treap上进行操作,然后返回结果。对于不整的区间,递归解决左半部分和右半部分,然后加起来返回。
第二种操作:对于一整段的区间可以直接在Treap上找区间第K大,但是对于不整的区间就无能为力了。所以没办法只能在套一层logn,二分查找这个值。对于二分的mid值,查找这个值在区间内的排名,这显然满足二分性质。要好好讨论一下这个问的边界条件。
第三种操作:递归向下修改,没经过一个线段树的节点,就把这个节点下面接的Treap减去原来位置的那个值,然后加上那个位置变成的值。
最后两种操作:对于整区间,直接返回Treap返回的前驱,后继。不整的区间,递归求出左半部分,右半部分,取最小或最大值,然后返回。
CODE:
#include <cstdio> #include <cstring> #include <iostream> #include <algorithm> #define MAX 50010 #define INF 1e9 #define LEFT (pos << 1) #define RIGHT (pos << 1|1) #define SIZE(x) ((x == NULL) ? 0:x->size) using namespace std; struct Complex{ int val,cnt,size,random; Complex *son[2]; Complex() { son[0] = son[1] = NULL; cnt = size = 1; random = rand(); } int Compare(int x) { if(x == val) return -1; return x > val; } void Maintain() { size = cnt; if(son[0] != NULL) size += son[0]->size; if(son[1] != NULL) size += son[1]->size; } }; int cnt,asks; int src[MAX]; Complex *tree[MAX << 2]; void Pretreatment(); void BuildTree(int l,int r,int pos); int GetRank(int l,int r,int x,int y,int pos,int k); int GetKth(int x,int y,int pos,int k); void Modify(int l,int r,int aim,int pos,int c); int FindPred(int l,int r,int x,int y,int pos,int k); int FindSucc(int l,int r,int x,int y,int pos,int k); inline void Rotate(Complex *&a,bool dir); void Insert(Complex *&a,int x); void Delete(Complex *&a,int x); int GetRank(Complex *a,int k); int FindPred(Complex *a,int x); int FindSucc(Complex *a,int x); int main() { Pretreatment(); cin >> cnt >> asks; for(int i = 1;i <= cnt; ++i) scanf("%d",&src[i]); BuildTree(1,cnt,1); for(int flag,i = 1;i <= asks; ++i) { scanf("%d",&flag); int x,y,z; if(flag == 1) { scanf("%d%d%d",&x,&y,&z); printf("%d\n",GetRank(1,cnt,x,y,1,z) + 1); } if(flag == 2) { scanf("%d%d%d",&x,&y,&z); printf("%d\n",GetKth(x,y,1,z)); } if(flag == 3) { scanf("%d%d",&x,&y); Modify(1,cnt,x,1,y); src[x] = y; } if(flag == 4) { scanf("%d%d%d",&x,&y,&z); printf("%d\n",FindPred(1,cnt,x,y,1,z)); } if(flag == 5) { scanf("%d%d%d",&x,&y,&z); printf("%d\n",FindSucc(1,cnt,x,y,1,z)); } } return 0; } void Pretreatment() { memset(tree,NULL,sizeof(tree)); } void BuildTree(int l,int r,int pos) { for(int i = l;i <= r; ++i) Insert(tree[pos],src[i]); if(l == r) return ; int mid = (l + r) >> 1; BuildTree(l,mid,LEFT); BuildTree(mid + 1,r,RIGHT); } int GetRank(int l,int r,int x,int y,int pos,int k) { if(l == x && y == r) return GetRank(tree[pos],k); int mid = (l + r) >> 1; if(y <= mid) return GetRank(l,mid,x,y,LEFT,k); if(x > mid) return GetRank(mid + 1,r,x,y,RIGHT,k); int left = GetRank(l,mid,x,mid,LEFT,k); int right = GetRank(mid + 1,r,mid + 1,y,RIGHT,k); return left + right; } int GetKth(int x,int y,int pos,int k) { int L = 0,R = INF,re; while(L <= R) { int mid = (L + R) >> 1; int temp = GetRank(1,cnt,x,y,1,mid); if(temp < k) L = mid + 1; else R = mid - 1; } int temp = GetRank(1,cnt,x,y,1,L); if(temp >= k) L = FindPred(1,cnt,x,y,1,L); return L; } void Modify(int l,int r,int aim,int pos,int c) { Delete(tree[pos],src[aim]); Insert(tree[pos],c); if(l == r) return ; int mid = (l + r) >> 1; if(aim <= mid) Modify(l,mid,aim,LEFT,c); else Modify(mid + 1,r,aim,RIGHT,c); } int FindPred(int l,int r,int x,int y,int pos,int k) { if(l == x && y == r) return FindPred(tree[pos],k); int mid = (l + r) >> 1; if(y <= mid) return FindPred(l,mid,x,y,LEFT,k); if(x > mid) return FindPred(mid + 1,r,x,y,RIGHT,k); int left = FindPred(l,mid,x,mid,LEFT,k); int right = FindPred(mid + 1,r,mid + 1,y,RIGHT,k); return max(left,right); } int FindSucc(int l,int r,int x,int y,int pos,int k) { if(l == x && y == r) return FindSucc(tree[pos],k); int mid = (l + r) >> 1; if(y <= mid) return FindSucc(l,mid,x,y,LEFT,k); if(x > mid) return FindSucc(mid + 1,r,x,y,RIGHT,k); int left = FindSucc(l,mid,x,mid,LEFT,k); int right = FindSucc(mid + 1,r,mid + 1,y,RIGHT,k); return min(left,right); } ////////////////////////////////Treap/////////////////////////// inline void Rotate(Complex *&a,bool dir) { Complex *k = a->son[!dir]; a->son[!dir] = k->son[dir]; k->son[dir] = a; a->Maintain(),k->Maintain(); a = k; } void Insert(Complex *&a,int x) { if(a == NULL) { a = new Complex(); a->val = x; return ; } int dir = a->Compare(x); if(dir == -1) a->cnt++; else { Insert(a->son[dir],x); if(a->son[dir]->random > a->random) Rotate(a,!dir); } a->Maintain(); } void Delete(Complex *&a,int x) { int dir = a->Compare(x); if(dir != -1) Delete(a->son[dir],x); else { if(a->cnt > 1) a->cnt--; else { if(a->son[0] == NULL) a = a->son[1]; else if(a->son[1] == NULL) a = a->son[0]; else { bool _dir = (a->son[0]->random > a->son[1]->random); Rotate(a,_dir); Delete(a->son[_dir],x); } } } if(a != NULL) a->Maintain(); } int GetRank(Complex *a,int k) { if(a == NULL) return 0; if(k <= a->val) return GetRank(a->son[0],k); return SIZE(a->son[0]) + a->cnt + GetRank(a->son[1],k); } int FindPred(Complex *a,int x) { if(a == NULL) return -INF; if(a->val >= x) return FindPred(a->son[0],x); return max(a->val,FindPred(a->son[1],x)); } int FindSucc(Complex *a,int x) { if(a == NULL) return INF; if(a->val <= x) return FindSucc(a->son[1],x); return min(a->val,FindSucc(a->son[0],x)); }
BZOJ 3196 二逼平衡树 树套树(线段树套Treap)
声明:以上内容来自用户投稿及互联网公开渠道收集整理发布,本网站不拥有所有权,未作人工编辑处理,也不承担相关法律责任,若内容有误或涉及侵权可进行投诉: 投诉/举报 工作人员会在5个工作日内联系你,一经查实,本站将立刻删除涉嫌侵权内容。