P2617 Dynamic Rankings（主席树模板题）

题目链接

题目大意：

对于一个长度为n的数列a，你需要高效地维护如下两种操作：
1、修改，将某个数a[i]改为指定的值t
2、查询，询问区间[i,j]中第k小的数字是多少

---

前置知识：树状数组，权值线段树

动态找第k小。

树套树的题中，最外的树一般是树状数组，复杂度必线段树低，而且好写。
这种情况下，最外层一般是单点修改（如，这题是修改指定值），里面可以用区间查询

但是如果外面也要区间修改，树状数组也可以写，但写法变得复杂，（支持区间修改的树状数组比较繁琐），不比线段树套线段树有章法。
和这篇是线段树套线段树的题做比较：bzoj1513 [POI2006]Tet-Tetris 3D（二维线段树，永久标记和延迟标记）

第二遍写这题，还是不熟练，多刷题吧。

---

完整代码：

#include <bits/stdc++.h>
//#pragma GCC optimize(2)
using namespace std;
typedef long long ll;
const int inf=0x3f3f3f3f;
const int mod=1e9+7;
int read(){
    int x=0,f=1;
    char ch=getchar();
    while(ch<'0'||ch>'9'){
        if(ch=='-')f=-1;
        ch=getchar();
    }
    while(ch>='0'&&ch<='9'){
        x=(x<<1)+(x<<3)+ch-'0';
        ch=getchar();
    }
    return x*f;
}
char getch(){
	char ch=getchar();
	while(ch!='Q' && ch!='C')	ch=getchar();
	return ch;
}
const int maxn=2e5+7;
int n,m,cnt,c,a[maxn],b[maxn<<1],t[maxn];
int rt[maxn*400],L[maxn*400],R[maxn*400],sum[maxn*400];
int op[maxn],x[maxn],y[maxn],z[maxn];
int lc,rc,tl[maxn],tr[maxn];
int lowbit(int x){	return x&(-x);	}
void update(int &i,int pre,int l,int r,int x,int v){
	i=++cnt;	L[i]=L[pre];	R[i]=R[pre];
	sum[i]=sum[pre]+v;
	if(l==r)	return;
	int mid=l+r>>1;
	if(x<=mid)	update(L[i],L[pre],l,mid,x,v);
	else		update(R[i],R[pre],mid+1,r,x,v);
}
int query(int l,int r,int k){
	if(l==r)	return l;
	int mid=l+r>>1,ans=0;
	for(int i=1;i<=lc;i++)	ans-=sum[L[tl[i]]];
	for(int i=1;i<=rc;i++)	ans+=sum[L[tr[i]]];
	if(k<=ans){
		for(int i=1;i<=lc;i++)	tl[i]=L[tl[i]];
		for(int i=1;i<=rc;i++)	tr[i]=L[tr[i]];
		return query(l,mid,k);
	}
	else{
		for(int i=1;i<=lc;i++)	tl[i]=R[tl[i]];
		for(int i=1;i<=rc;i++)	tr[i]=R[tr[i]];
		return query(mid+1,r,k-ans);
	}
}
void modify(int x,int v){
	int k=lower_bound(b,b+c,a[x])-b;
	for(int i=x;i<=n;i+=lowbit(i))	update(rt[i],rt[i],0,c-1,k,v);
}
int ask(int l,int r,int k){
	lc=rc=0;
	for(int i=l-1;i;i-=lowbit(i))	tl[++lc]=rt[i];
	for(int i=r;i;i-=lowbit(i))		tr[++rc]=rt[i];
	return query(0,c-1,k);
}
int main(){
	c=n=read();	m=read();
	for(int i=1;i<=n;i++)	b[i-1]=a[i]=read();
	for(int i=0;i<m;i++){
		char ch=getch();	x[i]=read();	y[i]=read();
		if(ch=='C'){
			op[i]=1;
			b[c++]=y[i];
		}
		else	z[i]=read();
	}
	sort(b,b+c);	c=unique(b,b+c)-b;
	for(int i=1;i<=n;i++)	modify(i,1);
	for(int i=0;i<m;i++){
		if(op[i]){
			modify(x[i],-1);
			a[x[i]]=y[i];
			modify(x[i],1);
		}
		else	printf("%d\n",b[ask(x[i],y[i],z[i])]);
	}
}
/*
5 3
3 2 1 4 7
Q 1 4 3
C 2 6
Q 2 5 3


3
6
*/