题目描述
输入
输出
对于每一次询问,你都需要输出他的答案,每一个输出占单独的一行。
样例输入
3 2 1 4 7
Q 1 4 3
C 2 6
Q 2 5 3
样例输出
6
带修改主席树经典题,树状数组上每个点建一棵主席树,存树状数组上这个点包含的序列中点的信息,修改看成删除和插入,每次操作跳lowbit。
#include<set>
#include<map>
#include<queue>
#include<cmath>
#include<stack>
#include<vector>
#include<cstdio>
#include<cstring>
#include<iostream>
#include<algorithm>
typedef long long ll;
using namespace std;
int n,m;
char ch[2];
int x,y,z;
int cnt;
int tot;
int num;
int ls[8000010];
int rs[8000010];
int sum[8000010];
int root[100010];
int s[100010];
int t[100010];
int a[100010];
int updata(int pre,int l,int r,int k)
{int rt=++cnt;if(l==r){sum[rt]=sum[pre]+1;return rt;}ls[rt]=ls[pre];rs[rt]=rs[pre];sum[rt]=sum[pre]+1;int mid=(l+r)>>1;if(k<=mid){ls[rt]=updata(ls[pre],l,mid,k);}else{rs[rt]=updata(rs[pre],mid+1,r,k);}return rt;
}
int downdata(int pre,int l,int r,int k)
{int rt=++cnt;if(l==r){sum[rt]=sum[pre]-1;return rt;}ls[rt]=ls[pre];rs[rt]=rs[pre];sum[rt]=sum[pre]-1;int mid=(l+r)>>1;if(k<=mid){ls[rt]=downdata(ls[pre],l,mid,k);}else{rs[rt]=downdata(rs[pre],mid+1,r,k);}return rt;
}
void add(int v,int x)
{for(int i=x;i<=n;i+=i&-i){root[i]=updata(root[i],0,1e9,v);}
}
void del(int v,int x)
{for(int i=x;i<=n;i+=i&-i){root[i]=downdata(root[i],0,1e9,v);}
}
int query(int l,int r,int k)
{if(l==r){return l;}int ans=0;int res=0;for(int i=1;i<=num;i++){res+=sum[ls[s[i]]];}for(int i=1;i<=tot;i++){ans+=sum[ls[t[i]]];}int mid=(l+r)>>1;if(ans-res>=k){for(int i=1;i<=num;i++){s[i]=ls[s[i]];}for(int i=1;i<=tot;i++){t[i]=ls[t[i]];}return query(l,mid,k);}else{for(int i=1;i<=num;i++){s[i]=rs[s[i]];}for(int i=1;i<=tot;i++){t[i]=rs[t[i]];}return query(mid+1,r,k-(ans-res));}
}
void ask(int l,int r,int x)
{num=0;tot=0;for(int i=l;i;i-=i&-i){s[++num]=root[i];}for(int i=r;i;i-=i&-i){t[++tot]=root[i];}printf("%d\n",query(0,1e9,x));
}
int main()
{scanf("%d%d",&n,&m);for(int i=1;i<=n;i++){scanf("%d",&a[i]);add(a[i],i);}for(int i=1;i<=m;i++){scanf("%s",ch);if(ch[0]=='Q'){scanf("%d%d%d",&x,&y,&z);ask(x-1,y,z);}else{scanf("%d%d",&x,&y);del(a[x],x);a[x]=y;add(a[x],x);}}
}