题目分析:
显然孤零零的线段树并不能支持插入之后的区间查询。
那么就需要一个支持插入的平衡树来帮忙了。
每个平衡树节点上开一个权值线段树,平衡树维护排名,线段树查询第K大
但是要维护子树信息就不能旋转,那就替罪羊树了。
节点上的线段树是当前节点的值和儿子的线段树的合并
(然而我写的线段树合并莫名TLE果然还是太菜了。。所以就改成暴力插入。。)
查询的时候就从平衡树的根下去,根据线段树的权值个数来划分区间,把线段树取出来,然后外面二分求第K小。
平衡树一个log,线段树一个log,重构的时候由于是暴力插入所以是两个log,总的复杂度应该在nlog3n少一些。
打完又是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);}
}
}