bzoj 3196 tyvj 1730 2 강 평형 수

3196: Tyvj 1730 이 강 평형 수
Time Limit: 10 Sec  
Memory Limit: 128 MB
Submit: 1862  
Solved: 789
[ Submit][ Status][ Discuss]
Description
데이터 구조 (제목 참조) 를 써 서 질서 있 는 수열 을 유지 해 야 합 니 다. 그 중에서 다음 과 같은 조작 을 제공 해 야 합 니 다.(후계 정 의 는 x 보다 크 고 가장 작은 수)
Input
첫 번 째 줄 두 개 수 n, m 는 길이 n 의 질서 있 는 서열 과 m 개의 조작 두 번 째 줄 은 n 개의 수 를 나타 내 고 질서 있 는 서열 아래 에 m 줄 이 있다 는 것 을 나타 낸다. opt 는 조작 레이 블 이 opt = 1 이면 조작 1 이 고 그 다음 에 세 개의 수 l, r, k 는 k 가 구간 [l, r] 의 순 위 를 조회 하면 opt = 2 는 조작 2 이 고 그 다음 에 세 개의 수 l, r, k 는 조회 구간 [l, r] 을 나타 낸다.내 순 위 는 k 의 수 는 opt = 3 이면 조작 3 이 고 그 다음 에 두 개의 수 pos 가 있다. k 는 pos 위치의 수 를 k 약 opt = 4 로 수정 하면 조작 4 이 고 그 다음 에 세 개의 수 l 이 있다. r, k 는 조회 구간 [l, r] 내 k 의 전구 인 opt = 5 는 조작 5 이 고 그 다음 에 세 개의 수 l, r, k 는 조회 구간 [l, r] 내 k 의 후계 이다.
Output
조작 1, 2, 4, 5 각 출력 한 줄 에 대해 조회 결 과 를 표시 합 니 다.
Sample Input
9 6 4 2 2 1 9 4 0 1 1 2 1 4 3 3 4 10 2 1 4 3 1 2 5 9 4 3 9 5 5 2 8 5
Sample Output
2 4 3 4 9
HINT
1. n 과 m 의 데이터 범위: n, m < = 50000
2. 시퀀스 의 각 수의 데이터 범위: [0, 1e8]
3. 원 제 는 없 지만 사실상 5 조작 k 는 마이너스 일 수 있 습 니 다.
Source
선분 트 리 커버 밸 런 스 트 리 누 드 문제, 다 쓰 고 나 서 다 들 몰 아 붙 였 는데...
k 번 째 큰 숫자 를 찾 으 려 면 2 점... 나머지 는 treap 기본 동작 입 니 다.
코드 연습 을 많이 해 야 될 것 같은 데...

#include<iostream>
#include<cstdio>
#include<cmath>
#include<cstdlib>
#include<cstring>
#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 inf 1000000000
#define maxn 50005
#define maxm 2000000
using namespace std;
int n,m,tot,ans;
int a[maxn],l[maxm],r[maxm],rnd[maxm],v[maxm],w[maxm],s[maxm];
struct seg{int l,r,rt;}t[4*maxn];
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 void pushup(int k){s[k]=s[l[k]]+s[r[k]]+w[k];}
inline void rturn(int &k){int t=l[k];l[k]=r[t];r[t]=k;s[t]=s[k];pushup(k);k=t;}
inline void lturn(int &k){int t=r[k];r[k]=l[t];l[t]=k;s[t]=s[k];pushup(k);k=t;}
inline void ins(int &k,int x)
{
	if (!k){k=++tot;v[k]=x;s[k]=w[k]=1;l[k]=r[k]=0;rnd[k]=rand();return;}
	s[k]++;
	if (v[k]==x) w[k]++;
	else if (x<v[k]){ins(l[k],x);if (rnd[l[k]]<rnd[k]) rturn(k);}
	else{ins(r[k],x);if (rnd[r[k]]<rnd[k]) lturn(k);}
}
inline void del(int &k,int x)
{
	if (v[k]==x)
	{
		if (w[k]>1){w[k]--;s[k]--;return;}
		if (l[k]*r[k]==0) k=l[k]+r[k];
		else if (rnd[l[k]]<rnd[r[k]]){rturn(k);del(k,x);}
		else{lturn(k);del(k,x);}
	}
	else if (x<v[k]){s[k]--;del(l[k],x);}
	else{s[k]--;del(r[k],x);}
}
inline void build(int k,int x,int y)
{
	int l=t[k].l=x,r=t[k].r=y,mid=(l+r)>>1;
	F(i,l,r) ins(t[k].rt,a[i]);
	if (l==r) return;
	build(k<<1,l,mid);build(k<<1|1,mid+1,r);
}
inline int rank(int k,int x)
{
	if (!k) return 0;
	if (v[k]==x) return s[l[k]];
	else if (x<v[k]) rank(l[k],x);
	else return s[l[k]]+w[k]+rank(r[k],x);
}
inline int pre(int k,int x)
{
	if (!k) return -inf;
	else if (x<=v[k]) return pre(l[k],x);
	else{int tmp=pre(r[k],x);return tmp==-inf?v[k]:tmp;}
}
inline int suc(int k,int x)
{
	if (!k) return inf;
	else if (x>=v[k]) suc(r[k],x);
	else{int tmp=suc(l[k],x);return tmp==inf?v[k]:tmp;}
}
inline void change(int k,int x,int y)
{
	int l=t[k].l,r=t[k].r,mid=(l+r)>>1;
	del(t[k].rt,a[x]);ins(t[k].rt,y);
	if (l==r) return;
	if (x<=mid) change(k<<1,x,y);
	else change(k<<1|1,x,y);
}
inline int query(int k,int x,int y,int z)
{
	int l=t[k].l,r=t[k].r,mid=(l+r)>>1;
	if (l==x&&r==y) return rank(t[k].rt,z);
	if (y<=mid) return query(k<<1,x,y,z);
	else if (x>mid) return query(k<<1|1,x,y,z);
	else return query(k<<1,x,mid,z)+query(k<<1|1,mid+1,y,z);
}
inline int getpre(int k,int x,int y,int z)
{
	int l=t[k].l,r=t[k].r,mid=(l+r)>>1;
	if (l==x&&r==y) return pre(t[k].rt,z);
	if (y<=mid) return getpre(k<<1,x,y,z);
	else if (x>mid) return getpre(k<<1|1,x,y,z);
	else return max(getpre(k<<1,x,mid,z),getpre(k<<1|1,mid+1,y,z));
}
inline int getsuc(int k,int x,int y,int z)
{
	int l=t[k].l,r=t[k].r,mid=(l+r)>>1;
	if (l==x&&r==y) return suc(t[k].rt,z);
	if (y<=mid) return getsuc(k<<1,x,y,z);
	else if (x>mid) return getsuc(k<<1|1,x,y,z);
	else return min(getsuc(k<<1,x,mid,z),getsuc(k<<1|1,mid+1,y,z));
}
int main()
{
	n=read();m=read();
	F(i,1,n) a[i]=read();
	build(1,1,n);
	int x,y,z,f;
	while (m--)
	{
		f=read();
		if (f==3){x=read();y=read();change(1,x,y);a[x]=y;}
		else
		{
			x=read();y=read();z=read();
			if (f==2)
			{
				int l=0,r=inf;
				while (l<=r)
				{
					int mid=(l+r)>>1;
					if (query(1,x,y,mid)+1>z) r=mid-1;
					else l=mid+1;
				}
				printf("%d
",r);
			}
			if (f==1) printf("%d
",query(1,x,y,z)+1);
			if (f==4) printf("%d
",getpre(1,x,y,z));
			if (f==5) printf("%d
",getsuc(1,x,y,z));
		}
	}
	return 0;
}