[线段树套平衡树] BZOJ 2883 gss2加强版

每个数支持出现一次 那么给每个数再赋一个权值pre

每次查询只要查[L,R]中pre<L的和 然后就是个树套树了

结果splay被卡了 打了个treap

#include<cstdio>
#include<cstdlib>
#include<algorithm>
#include<set>
#define N 100010
using namespace std;
typedef long long ll;

inline char nc()
{
	static char buf[100000],*p1=buf,*p2=buf;
	if (p1==p2) { p2=(p1=buf)+fread(buf,1,100000,stdin); if (p1==p2) return EOF; }
	return *p1++;
}

inline void read(int &x)
{
	char c=nc(),b=1;
	for (;!(c>='0' && c<='9');c=nc()) if (c=='-') b=-1;
	for (x=0;c>='0' && c<='9';x=x*10+c-'0',c=nc()); x*=b;
}

inline void read(char &x)
{
	for (x=nc();x!='Q' && x!='U';x=nc());
}

int n,m;
int a[N],eve[N][3],last[N<<1];
set<int>T[N<<1];

ll ans;

int sx[N<<1],icnt;

inline int Bin(int x){
	return lower_bound(sx+1,sx+icnt+1,x)-sx;
}

struct node{
  	int val,fix; ll v,sum;
	node *l,*r;
	void update(){ sum=v+l->sum+r->sum; }
}*null,*root[N<<2];

const int ND=2000000;
node Mem[ND],*Stack[ND];
int top;

inline void Init(){
	null=Mem; null->l=null->r=null; null->val=null->sum=0;
	for (int i=1;i<ND;i++) Stack[++top]=Mem+i;
}
inline node *New(){ return Stack[top--]; }
inline void Del(node *x) { Stack[++top]=x; }

inline void Rotatel(node*&x){ node*y=x->r;x->r=y->l;x->update();y->l=x;y->update();x=y;}
inline void Rotater(node*&x){ node*y=x->l;x->l=y->r;x->update();y->r=x;y->update();x=y;}

void Ins(node *&x,int p,int v)
{
	if(x==null){
	   x=New(); x->val=p; x->l=x->r=null; x->v=x->sum=v; x->fix=std::rand();
	   return;
	}
	x->sum+=v;
	if(p==x->val){ 
		x->v+=v; return;
	}else if(p<x->val){
		Ins(x->l,p,v);
	    if(x->l->fix>x->fix) Rotater(x);
	}else{
		Ins(x->r,p,v);
	    if(x->r->fix>x->fix) Rotatel(x);
	}
}
ll Ask(node *x,int p){
	if(x==null) return 0;
  	if(p==x->val) 
	  	return x->l->sum;
  	else if(p>x->val) 
	  	return x->v+x->l->sum+Ask(x->r,p);
  	else
	  	return Ask(x->l,p);
}

void Build(int x,int l,int r){
  	root[x]=null; int mid=(l+r)>>1;
  	if (l==r) return;
  	Build(x<<1,l,mid); Build(x<<1|1,mid+1,r);
}

inline void Add(int t,int p,int q){
	int x=1,l=1,r=n,mid;
	while(l<r){
	Ins(root[x],p,q);
	if(t<=(mid=(l+r)>>1))
		r=mid,x=x<<1;
	else 
		l=mid+1,x=x<<1|1;
	}
	Ins(root[x],p,q);
}

void Query(int x,int l,int r,int L,int R,int X)
{
  	if (L<=l && r<=R){
	  	ans+=Ask(root[x],X); return;
	}
  	int mid=(l+r)>>1;
  	if(L<=mid)
	  	Query(x<<1,l,mid,L,R,X);
  	if(R>mid)
	  	Query(x<<1|1,mid+1,r,L,R,X);
}

int main()
{
	char order; int pre,nxt,x,y;
	freopen("t.in","r",stdin);
	freopen("t.out","w",stdout);
	Init();
	read(n);
  	for (int i=1;i<=n;i++) 
		read(a[i]),sx[++icnt]=a[i];
	read(m);
  	for(int i=1;i<=m;i++){
    	read(order); read(eve[i][1]); read(eve[i][2]);
    	if(order=='U') sx[++icnt]=eve[i][2]; else eve[i][0]=1;
  	}
	sort(sx+1,sx+icnt+1); icnt=unique(sx+1,sx+icnt+1)-sx-1;
  	for(int i=1;i<=icnt;i++)
		T[i].insert(0),T[i].insert(n+1);
	Build(1,1,n);
	for(int i=1;i<=n;i++){
    	T[a[i]=Bin(a[i])].insert(i);
    	Add(i,last[a[i]],sx[a[i]]);
    	last[a[i]]=i;
  	}
	set<int>::iterator j,k;
  	for(int i=1;i<=m;i++)
		if(eve[i][0])
		{
			ans=0;
			Query(1,1,n,eve[i][1],eve[i][2],eve[i][1]);
			printf("%lld\n",ans);
 		}
  		else
		{
	    	x=eve[i][1],y=Bin(eve[i][2]);
	    	k=j=T[a[x]].find(x);
			k--; pre=*k; k=j; k++; nxt=*k;
	    	Add(x,pre,-sx[a[x]]);
	    	if(nxt<=n) 
				Add(nxt,x,-sx[a[x]]),Add(nxt,pre,sx[a[x]]);
	    	T[a[x]].erase(x); T[a[x]=y].insert(x);
	    	k=j=T[y].find(x);
			k--; pre=*k; k=j; k++; nxt=*k;
	    	Add(x,pre,sx[y]);
	    	if(nxt<=n) 
				Add(nxt,pre,-sx[y]),Add(nxt,x,sx[y]);
	  	}
  	return 0;
}