【题解】BZOJ 1901: Zju2112 Dynamic Rankings

题目传送门（权限题）

一道类似的非权限题

题意

N个数的序列，每次修改一个数或者询问区间里的第K小。可以离线。

简要做法

假如要求在线，可以树状数组套个线段树~

但是这题是可以离线的，就可以乱搞整体二分啦~

非常好写好调。（图文无关）

细节

• 可以把一个修改拆成+1和-1两个操作
• 数组要开到30000左右，如果初始数组也当做操作的话

代码

#include <bits/stdc++.h>
#define TR(x) cerr<<#x<<'='<<x<<endl
using namespace std;
const int MAXN=30005, INF=0x3f3f3f3f;
void rd(int &x){
	x=0; int ch=getchar();
	while(ch<'0'||'9'<ch) ch=getchar();
	while('0'<=ch&&ch<='9') x=x*10+ch-'0', ch=getchar();
}
int N, M, nq;
int A[MAXN], c[MAXN], id[MAXN], q1[MAXN], q2[MAXN], ans[MAXN];
void add(int x, int k){for(;x<=N;x+=x&-x)c[x]+=k;}
int sum(int x){int r=0;for(;x;x-=x&-x)r+=c[x];return r;}
struct Qry{
	int t,a,b,k;
	Qry(int t,int a,int b,int k):t(t),a(a),b(b),k(k){}
	Qry(){}
}Q[MAXN];
void solve(int *a, int n, int l, int r){
	if(!n) return;
	if(l==r){
		for(int i=0; i<n; ++i) ans[a[i]]=l;
		return;
	}
	int mid=(l+r)>>1, n1=0, n2=0;
	for(int i=0; i<n; ++i){
		Qry &q=Q[a[i]];
		if(q.t==0){
			if(q.b<=mid) add(q.a,q.k),q1[n1++]=a[i];
			else q2[n2++]=a[i];
		}else{
			int t=sum(q.b)-sum(q.a-1);
			if(q.k<=t) q1[n1++]=a[i];
			else q.k-=t,q2[n2++]=a[i];
		}
	}
	for(int i=0; i<n; ++i){
		Qry &q=Q[a[i]];
		if(q.t==0&&q.b<=mid) add(q.a,-q.k);
	}
	memcpy(a,q1,sizeof(int)*n1);
	memcpy(a+n1,q2,sizeof(int)*n2);
	solve(a,n1,l,mid);
	solve(a+n1,n2,mid+1,r);
}
int main(){
	rd(N),rd(M);
	for(int i=1; i<=N; ++i) rd(A[i]),Q[nq++]=Qry(0,i,A[i],1);
	for(int i=1; i<=M; ++i){
		char o[3]; scanf("%s",o);
		int a,b,k;
		if(o[0]=='Q'){
			rd(a),rd(b),rd(k);
			Q[nq++]=Qry(1,a,b,k);
		}else{
			rd(a),rd(k);
			Q[nq++]=Qry(0,a,A[a],-1);
			Q[nq++]=Qry(0,a,A[a]=k,1);
		}
	}
	for(int i=0; i<nq; ++i) id[i]=i;
	solve(id,nq,0,INF);
	for(int i=0; i<nq; ++i)if(Q[i].t)printf("%d\n",ans[i]);
	return 0;
}