BZOJ 3065: 带插入区间K小值 【替罪羊树套线段树】

题目分析：

显然孤零零的线段树并不能支持插入之后的区间查询。
那么就需要一个支持插入的平衡树来帮忙了。
每个平衡树节点上开一个权值线段树，平衡树维护排名，线段树查询第K大
但是要维护子树信息就不能旋转，那就替罪羊树了。

节点上的线段树是当前节点的值和儿子的线段树的合并
（然而我写的线段树合并莫名TLE果然还是太菜了。。所以就改成暴力插入。。）

查询的时候就从平衡树的根下去，根据线段树的权值个数来划分区间，把线段树取出来，然后外面二分求第K小。

平衡树一个log，线段树一个log，重构的时候由于是暴力插入所以是两个log，总的复杂度应该在nlog³n少一些。

打完又是3h过去了。。
回收空间的时候注意清零，查询的时候划分区间仔细一些。。
Code：

#include<cstdio>
#include<cctype>
#include<vector>
#include<algorithm>
#define LL long long
#define maxn 70005
#define alpha 0.75
using namespace std;
inline void read(int &a){
    char c;bool f=0;
	while(!isdigit(c=getchar())) if(c=='-') f=1;
    for(a=c-'0';isdigit(c=getchar());a=a*10+c-'0');
	if(f) a=-a;
}
int n,m,v[maxn],ans,dfn[maxn],sz;
int root,rt[maxn],lc[maxn],rc[maxn];
int *rep;
struct Segment_Tree{int l,r,s;}a[10000000];
vector<int>rec,p,t;
inline int newnode(){
	if(rec.empty()) return ++sz;
	int tmp=rec.back();rec.pop_back();
	return tmp;
}
void reclaim(int &x){
	if(!x) return;
	rec.push_back(x);
	reclaim(a[x].l),reclaim(a[x].r);
	a[x].s=0; x=0;
}
void insert(int &now,int l,int r,int val,int d){
	if(!now) now=newnode();
	if(!(a[now].s+=d)) {reclaim(now);return;}
	if(l==r) return;
	int mid=(l+r)>>1;
	if(val<=mid) insert(a[now].l,l,mid,val,d);
	else insert(a[now].r,mid+1,r,val,d);
}
/*TLE...???
void merge(int &now,int x,int y){
	if(!x&&!y) return;
	if(!now) now=newnode();
	a[now].s+=a[x].s+a[y].s;
	merge(a[now].l,a[x].l,a[y].l);
	merge(a[now].r,a[x].r,a[y].r);
}
*/
void build(int &now,int l,int r){
	int mid=(l+r)>>1; now=dfn[mid];
	if(l<mid) build(lc[now],l,mid-1);
	if(r>mid) build(rc[now],mid+1,r);
	for(int i=l;i<=r;i++) insert(rt[now],0,70000,v[dfn[i]],1);	
}
void pia(int &x){
	if(!x) return; reclaim(rt[x]);
	pia(lc[x]); p.push_back(x); pia(rc[x]);
	x=0;
}
void rebuild(int &x){
	p.clear(),pia(x);
	for(int i=p.size()-1;i>=0;i--) dfn[i+1]=p[i];
	build(x,1,p.size());
}
int modify(int now,int x,int val){
	insert(rt[now],0,70000,val,1);
	int tv,L=a[rt[lc[now]]].s;
	if(L+1==x) {tv=v[now];v[now]=val;}
	else if(x<=L) tv=modify(lc[now],x,val);
	else tv=modify(rc[now],x-L-1,val);
	insert(rt[now],0,70000,tv,-1);
	return tv;
}
void insert(int &now,int x,int val){
	if(!now){
		v[now=++n]=val;
		insert(rt[now],0,70000,val,1);
		return;
	}
	insert(rt[now],0,70000,val,1);
	int L=a[rt[lc[now]]].s;
	if(x<=L) insert(lc[now],x,val);
	else insert(rc[now],x-L-1,val);
	if(a[rt[now]].s*alpha<=max(a[rt[lc[now]]].s,a[rt[rc[now]]].s)) rep=&now;
}
void query(int now,int l,int r){
	int L=a[rt[lc[now]]].s,R=a[rt[now]].s;//R is whole now!
	if(l==1&&r==R) {t.push_back(rt[now]);return;}
	if(l<=L+1&&L+1<=r) p.push_back(v[now]);
	if(r<=L) query(lc[now],l,r);
	else if(l>L+1) query(rc[now],l-L-1,r-L-1);
	else{
		if(l<=L) query(lc[now],l,L);
		if(r>L+1) query(rc[now],1,r-L-1);
	}
}
int solve(int x,int y,int K){
	t.clear(),p.clear();
	query(root,x,y);
	int l=0,r=70000,s1=t.size(),s2=p.size();
	while(l<r){
		int mid=(l+r)>>1,sum=0;
		for(int i=0;i<s1;i++) sum+=a[a[t[i]].l].s;
		for(int i=0;i<s2;i++) sum+=(l<=p[i]&&p[i]<=mid);
		if(K<=sum){
			for(int i=0;i<s1;i++) t[i]=a[t[i]].l;
			r=mid;
		}
		else{
			for(int i=0;i<s1;i++) t[i]=a[t[i]].r;
			l=mid+1,K-=sum;
		}
	}
	return l;
}
int main()
{
	//freopen("H.in","r",stdin);
	read(n);
	for(int i=1;i<=n;i++) read(v[i]),dfn[i]=i;
	build(root,1,n);
	read(m);
	char op;int x,y,k;
	while(m--){
		while(!isalpha(op=getchar()));
		read(x),read(y),x^=ans,y^=ans;
		if(op=='Q') read(k),k^=ans,printf("%d\n",ans=solve(x,y,k));
		if(op=='M')	modify(root,x,y);
		if(op=='I') {rep=0,insert(root,x-1,y); if(rep) rebuild(*rep);}
	}
}