luogu P2617 Dynamic Rankings

背景：

$1G$的空间。
$hehe...$

题目传送门：

https://www.luogu.org/problemnew/show/P2617

题意：

维护一个支持修改某一个权值和求区间第$k$大的东西。

思路：

带修主席树模板。
用一个树状数组维护一下维护主席树的前缀和即可。
时空复杂度：$\Theta (n{\log }^{2}n)$。
我才不会告诉你要根据数据范围开数组。

代码：

#include<cstdio>
#include<cstring>
#include<algorithm>
#define INF 2e9
#define lowbit(x) ((x)&(-x))
using namespace std;
	int last[100010],ls[40000010],rs[40000010],root[100010],tot[40000010],d[100010];
	int n,m,len=0;
void change(int &now,int x,int y,int l,int r)
{
	if(!now) now=++len;
	tot[now]+=y;
	if(l==r) return;
	int mid=(l+r)>>1;
	if(x<=mid) change(ls[now],x,y,l,mid); else change(rs[now],x,y,mid+1,r);
}
void work(int id,int t)
{
	for(int i=id;i<=n;i+=lowbit(i))
		change(root[i],last[id],-1,0,INF);
	for(int i=id;i<=n;i+=lowbit(i))
		change(root[i],t,1,0,INF);
	last[id]=t;
}
int xx[100010],yy[100010];
int t1,t2;
int solve(int k,int l,int r)
{
	if(l==r) return l;
	int tmp=0;
	for(int i=1;i<=t2;i++)
		tmp+=tot[ls[yy[i]]];
	for(int i=1;i<=t1;i++)
		tmp-=tot[ls[xx[i]]];
	int mid=(l+r)>>1;
	if(k<=tmp)
	{
		for(int i=1;i<=t2;i++)
			yy[i]=ls[yy[i]];
		for(int i=1;i<=t1;i++)
			xx[i]=ls[xx[i]];
		return solve(k,l,mid);
	}
	else
	{
		for(int i=1;i<=t2;i++)
			yy[i]=rs[yy[i]];
		for(int i=1;i<=t1;i++)
			xx[i]=rs[xx[i]];
		return solve(k-tmp,mid+1,r);
	}
}
int query(int x,int y,int k)
{
	t1=t2=0;
	for(int i=x;i;i-=lowbit(i))
		xx[++t1]=root[i];
	for(int i=y;i;i-=lowbit(i))
		yy[++t2]=root[i];
	return solve(k,0,INF);
}
int main()
{
	char s[10];
	int x,y,z;
	scanf("%d %d",&n,&m);
	for(int i=1;i<=n;i++)
	{
		scanf("%d",&last[i]);
		for(int j=i;j<=n;j+=lowbit(j))
			change(root[j],last[i],1,0,INF);
	}
	for(int i=1;i<=m;i++)
	{
		scanf("%s",s+1);
		if(s[1]=='C')
		{
			scanf("%d %d",&x,&y);
			work(x,y);
		}
		else
		{
			scanf("%d %d %d",&x,&y,&z);
			printf("%d\n",query(x-1,y,z));
		}
	}
}