题意:
单点修改,区间K大
题解
不会主席树的我,用的是根据值域的整体分治。
对于修改,可以拆成两个操作,一个是撤销,一个是增加。
一共是三种操作
op=-1时,代表撤销,x代表位置;
op=0时,代表增加,在x位置插入y;
op>0时,代表查询,在[x,y]区间内第z大。(x=l,y=r,z=k)
再根据值域分治的时候,
当op=0且q[i].y<=mid时,就正常插入。但注意此时不管q[i].y的大小,都要a[q[i].x]=i,存一下实际序列的q[i].x位是q[i]。这是因为当之后有一个把序列a[e]位置修改为f的操作的时候,那么第一步需要先把a[e]删了,这时候就要判断a[e]<=mid,如果为真需要在树状数组上删除a[e];如果为假就没事。
另外值得思考的是,撤销和增加是放在lq还是rq。
在出现修改之前,a[l~r]是已经初始化过的。
面对“C i t”的时候,拆成q[j]={op=-1,x=i},q[j+1]={op=0,x=i,y=t}。
当a[i]<=mid,t<=mid时:q[j]和q[j+1]都放在lq。
当a[i]<=mid,t>mid时:在之后的值域小于mid的时候,a[i]仍然会被撤回,所以q[j]放在lq。而q[j+1]可以直接丢到rq,因为在执行后面值域>mid的分治时,对于j+1之后的操作,a[i]<=mid,q[j]这两步操作是无意义的,只有q[j+1]是有意义的,所以可以直接在i位置放上t,即只把q[j+1]放进rq。
当a[i]>=mid,t>=mid时:对于当前分治没有影响,q[j]和q[j+1]都放在rq。
当a[i]>=mid,t<mid时:对于当前分治,a[i]>=mid,q[j]的操作是无意义的,完全可以直接q[j+1],所以将a[i]初始化的操作、以及q[j]直接丢到rq,将q[j]当作a[i]初始化。
#include <cstdio>
#include <iostream>
#include <vector>
#include <cstring>
#include <string>
#include <queue>
#include <algorithm>
#include <cmath>
#include <set>
#include <map>
#include <cstdlib>
#define pb push_back
#define fir first
#define sec second
#define All(x) (x).begin(),(x).end()
#define ms(a,b) memset((a),(b),sizeof(a))
#define INF 0x3f3f3f3f
#define sp system("pause")
#define multi int T;scanf("%d",&T);while((T)--)
using namespace std;
typedef long long ll;
typedef double db;
const int N=2e4+5;
const int mod=10007;
const db pi=acos(-1.0);
int n,m;
struct node{
int op,x,y,z;
}q[N<<1],lq[N<<1],rq[N<<1];
int tr[N],ans[N],a[N];
inline int ask(int x){int res=0;for(int i=x;i>0;i-=i&-i) res+=tr[i]; return res;}
inline void add(int x,int val) {for(int i=x;i<=n;i+=i&-i) tr[i]+=val;}
inline void clear(int x){for(int i=x;i<=n;i+=i&-i) tr[i]=0;}
void solve(int lval,int rval,int l,int r){
if(l>r) return ;
if(lval==rval){
for(int i=l;i<=r;i++) if(q[i].op>0) ans[q[i].op]=lval;
return ;
}
int mid=(lval+rval)>>1,rt=0,lt=0;
for(int i=l;i<=r;i++){
if(q[i].op==0){
if(q[i].y<=mid){
add(q[i].x,1);
lq[++lt]=q[i];
}else rq[++rt]=q[i];
a[q[i].x]=i;//q[i].y>mid也要给a,可能这个位置经过多次改变,每次更新a,知道这一位到底>mid,还是<=mid
}else{
if(q[i].op==-1){
int pos=a[q[i].x];
if(q[pos].y<=mid) {
lq[++lt]=q[i];
add(q[pos].x,-1);
}else rq[++rt]=q[i];
}else{
int cnt=ask(q[i].y)-ask(q[i].x-1);
if(cnt>=q[i].z) lq[++lt]=q[i];
else{
q[i].z-=cnt;
rq[++rt]=q[i];
}
}
}
}
for(int i=l;i<=r;i++){
if(q[i].op==0&&q[i].y<=mid) clear(q[i].x);
// if(q[i].op==-1){//此时a存储的已经不是最早对应的了
// int pos=a[q[i].x];
// if(q[pos].y<=mid) add(q[pos].x,1);
// }
}
for(int i=1;i<=lt;i++) q[l+i-1]=lq[i];
for(int i=1;i<=rt;i++) q[l+lt+i-1]=rq[i];
solve(lval,mid,l,l+lt-1);
solve(mid+1,rval,l+lt,r);
}
int main()
{
#ifndef ONLINE_JUDGE
freopen("D:\\work\\data.in","r",stdin);
#endif
scanf("%d%d",&n,&m);
for(int i=1;i<=n;i++){
scanf("%d",&q[i].y);
q[i].x=i;
q[i].op=0;
}
char op[5];
int t=n,anst=0;
for(int i=1;i<=m;i++){
++t;
scanf("%s%d%d",op,&q[t].x,&q[t].y);
if(op[0]=='Q'){
scanf("%d",&q[t].z);
q[t].op=++anst;
}else{
q[t].op=-1;
++t;
q[t].op=0;
q[t].x=q[t-1].x;
q[t].y=q[t-1].y;
}
}
solve(-INF,INF,1,t);
for(int i=1;i<=anst;i++) printf("%d\n",ans[i]);
}