[Bzoj1901]Zju2112 Dynamic Rankings

/*1.change后要更新a数组
  2.删除节点后子节点数量-1*/
#include<iostream>
#include<cstring>
#include<cstdlib>
#include<cstdio>
#define N 200010
#define M 1300010
#define inf 1000000000
using namespace std;
struct data{
	int l,r,s,w,v,rnd;
}tr[M];
int root[N],a[50001],n,m,sz,tmp; 
void update(int k){
	tr[k].s=tr[tr[k].l].s+tr[tr[k].r].s+tr[k].w;
}
void lturn(int &k){
	int t=tr[k].r;
	tr[k].r=tr[t].l;
	tr[t].l=k;
	tr[t].s=tr[k].s;
	update(k);k=t;	
}
void rturn(int &k){
	int t=tr[k].l;
	tr[k].l=tr[t].r;
	tr[t].r=k;
	tr[t].s=tr[k].s;
	update(k);k=t;
}
void insert(int &k,int num){
	if(!k){
		k=++sz;
		tr[k].s=tr[k].w=1;
		tr[k].rnd=rand();
		tr[k].l=tr[k].r=0;
		tr[k].v=num;
		return;
	}
	tr[k].s++;
	if(tr[k].v==num)tr[k].w++;
	else if(num<tr[k].v){
		insert(tr[k].l,num);
		if(tr[tr[k].l].rnd<tr[k].rnd)rturn(k);
	}
	else{
		insert(tr[k].r,num);
		if(tr[tr[k].r].rnd<tr[k].rnd)lturn(k);
	}
}
void del(int &k,int num){
	if(tr[k].v==num){
		if(tr[k].w>1){tr[k].w--;tr[k].s--;return;}
		else if(tr[k].l*tr[k].r==0)k=tr[k].l+tr[k].r;
		else if(tr[tr[k].l].rnd<tr[tr[k].r].rnd){
			rturn(k);del(k,num);
		}
		else{
			lturn(k);del(k,num);
		}
	}
	else if(num<tr[k].v){
		del(tr[k].l,num);tr[k].s--;
	}
	else{
		del(tr[k].r,num);tr[k].s--;
	}
}
void change(int k,int l,int r,int x,int num,int y){
	del(root[k],y);
	insert(root[k],num);
	if(l==r)return;
	int mid=(l+r)>>1;
	if(x<=mid)change(k<<1,l,mid,x,num,y);
	else change(k<<1|1,mid+1,r,x,num,y);
}
void build(int k,int l,int r,int x,int num){
	insert(root[k],num);
	if(l==r)return;
	int mid=(l+r)>>1;
	if(x<=mid)build(k<<1,l,mid,x,num);
	else build(k<<1|1,mid+1,r,x,num);
}
void get_rank(int k,int num){
	if(!k)return;
	if(num>=tr[k].v){
		tmp+=tr[tr[k].l].s+tr[k].w;
		get_rank(tr[k].r,num);
	}
	else get_rank(tr[k].l,num);
}
void ask_rank(int k,int l,int r,int x,int y,int num){
	if(l==x&&r==y){get_rank(root[k],num);return;}
	int mid=(l+r)>>1;
	if(y<=mid)ask_rank(k<<1,l,mid,x,y,num);
	else if(x>mid)ask_rank(k<<1|1,mid+1,r,x,y,num);
	else{
		ask_rank(k<<1,l,mid,x,mid,num);
		ask_rank(k<<1|1,mid+1,r,mid+1,y,num);
	}
}
void ask_index(int x,int y,int z){
    int l=0,r=inf,ans;
    while(l<=r)
    {
        int mid=(l+r)>>1;
        tmp=1;ask_rank(1,1,n,x,y,mid);
        if(tmp<=z){l=mid+1;ans=mid;}
        else r=mid-1;
        }
    printf("%d\n",l);
}
int main(){
		scanf("%d%d",&n,&m);
		for(int i=1;i<=n;i++)scanf("%d",&a[i]);
		for(int i=1;i<=n;i++)build(1,1,n,i,a[i]);
		for(int i=1;i<=m;i++){
			char s[3];int x,y,z;
			scanf("%s",s);
			if(s[0]=='Q'){
				scanf("%d%d%d",&x,&y,&z);
				ask_index(x,y,z);
			}
			else if(s[0]=='C'){
				scanf("%d%d",&x,&y);
				change(1,1,n,x,y,a[x]);
				a[x]=y;
			}
		}
	return 0;
}

发布了377 篇原创文章 · 获赞 2 · 访问量 12万+
展开阅读全文

没有更多推荐了,返回首页

©️2019 CSDN 皮肤主题: 大白 设计师: CSDN官方博客

分享到微信朋友圈

×

扫一扫,手机浏览