bzoj1901 Zju112 Dynamic Rankings

1901: Zju2112 Dynamic Rankings
Time Limit: 10 Sec  
Memory Limit: 128 MB
Submit: 6823  
Solved: 2840
[ Submit][ Status][ Discuss]
Description
n개수를 포함하는 서열 a[1], a[2], a[3]...a[n]를 정하면 프로그램은 반드시 이런 질문에 대답해야 한다. 정해진 i, j, k에 대해 a[i], a[i+1], a[i+2]......a[j]에서 작은 k가 얼마(1≤k≤j-i+1)인지, 그리고 a[i]의 값을 바꿀 수 있다. 바꾸면 프로그램은 바뀐 a에 대해 계속 위의 질문에 대답할 수 있다.입력 파일에서 시퀀스 a를 읽고, 질문 명령과 수정 명령을 포함한 일련의 명령을 읽는 프로그램을 만들어야 한다.모든 질문 지령에 대해, 너는 반드시 정확한 대답을 출력해야 한다.첫 번째 줄에는 두 개의 정수 n(1≤n≤10000), m(1≤m≤10000)이 있다.서열의 길이와 명령의 개수를 각각 나타낸다.두 번째 줄에는 n개의 수가 있는데 a[1], a[2]......a[n]을 나타낸다. 이 수는 모두 10^9보다 작다.다음 m행은 각 명령을 설명하고 각 행의 형식은 다음 두 가지 형식 중의 하나이다.Q i jk 또는 C i t Q i jk(i, j, k는 숫자, 1≤i ≤ j≤ n, 1≤ k≤ j-i+1)는 문의 지령을 표시하고, a[i], a[i+1]......a[j]에서 k가 작은 수를 묻는다.Cit(1≤i≤n, 0≤t≤10^9)는 a[i]를 t로 바꾸는 것을 의미한다.
Input
매번 질문할 때마다, 너는 그의 답안을 출력해야 한다. 모든 출력은 단독 줄을 차지한다.
Output
Sample Input
5 3 3 2 1 4 7 Q 1 4 3 C 2 6 Q 2 5 3
Sample Output
3 6
HINT
20%의 데이터 중 m, n≤100;40%의 데이터 중 m, n≤1000;100% 데이터 중 m, n≤10000.
수정된 세그먼트 K는 크고, 트리 배열 세트는 세그먼트 트리를 영구화합니다.
복잡도 O (n*log^2n).

#include<cmath>
#include<cstdio>
#include<cstdlib>
#include<cstring>
#include<iostream>
#include<algorithm>
#define F(i,j,n) for(int i=j;i<=n;i++)
#define D(i,j,n) for(int i=j;i>=n;i--)
#define ll long long
#define N 20001
#define M 5000000
using namespace std;
int n,m,cnt,tot;
int a[N],p[N],f[N],rt[N],L[N],R[N];
int sz[M],ls[M],rs[M];
struct opt{int x,y,z,flg;}q[N];
inline 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*10+ch-'0';ch=getchar();}
	return x*f;
}
inline int find(int x)
{
	int l=1,r=tot,mid;
	while (l<=r)
	{
		mid=(l+r)>>1;
		if (f[mid]==x) return mid;
		if (f[mid]<x) l=mid+1;
		else r=mid-1;
	}
}
void insert(int x,int &y,int l,int r,int v,int w)
{
	y=++cnt;sz[y]=sz[x]+w;
	if (l==r) return;
	int mid=(l+r)>>1;
	if (v<=mid) rs[y]=rs[x],insert(ls[x],ls[y],l,mid,v,w);
	else ls[y]=ls[x],insert(rs[x],rs[y],mid+1,r,v,w);
}
int query(int l,int r,int k)
{
	int num,mid;
	while (l<r)
	{
		num=0;mid=(l+r)>>1;
		F(i,1,R[0]) num+=sz[ls[R[i]]];
		F(i,1,L[0]) num-=sz[ls[L[i]]];
		if (num>=k)
		{
			r=mid;
			F(i,1,L[0]) L[i]=ls[L[i]];
			F(i,1,R[0]) R[i]=ls[R[i]];
		}
		else
		{
			l=mid+1;k-=num;
			F(i,1,L[0]) L[i]=rs[L[i]];
			F(i,1,R[0]) R[i]=rs[R[i]];
		}
	}
	return l;
}
int main()
{
	n=read();m=read();
	F(i,1,n) p[++cnt]=a[i]=read();
	F(i,1,m)
	{
		char ch=getchar();while (ch!='C'&&ch!='Q') ch=getchar();
		if (ch=='Q'){q[i].flg=1;q[i].x=read()-1;q[i].y=read();q[i].z=read();}
		else{q[i].flg=2;q[i].x=read();p[++cnt]=q[i].y=read();}
	}
	sort(p+1,p+cnt+1);
	F(i,1,cnt) if (i==1||p[i]!=p[i-1]) f[++tot]=p[i];
	cnt=0;
	F(i,1,n)
	{
		a[i]=find(a[i]);
		for(int j=i;j<=n;j+=(j&(-j))) insert(rt[j],rt[j],1,tot,a[i],1);
	}
	F(i,1,m)
	{
		if (q[i].flg==1)
		{
			L[0]=R[0]=0;
			for(int j=q[i].x;j;j-=(j&(-j))) L[++L[0]]=rt[j];
			for(int j=q[i].y;j;j-=(j&(-j))) R[++R[0]]=rt[j];
			printf("%d
",f[query(1,tot,q[i].z)]);
		}
		else
		{
			int x=q[i].x,y=find(q[i].y);
			for(int j=x;j<=n;j+=(j&(-j))) insert(rt[j],rt[j],1,tot,a[x],-1),insert(rt[j],rt[j],1,tot,y,1);
			a[x]=y;
		}
	}
	return 0;
}