JZOJ 5933. 【NOIP2018模拟10.27】百鸽笼

Description

Input

从文件 pigeon.in 中读入数据。
输入第一行包含两个正整数 n, m ，分别表示初始鸽笼数与操作个数。
第二行包含 n 个正整数，第 i 个数表示从左往右第 i 个初始鸽笼中鸽子的咕咕能力值 vi 。
接下去 m 行每行表示一个操作。操作输入格式见题面描述。

Output

输出到文件 pigeon.out 中。
输出包含若干行，每行表示一个相应的 3 操作的答案。

Sample Input

6 8
2 7 4 3 5 9
3 2 5 3
1
2 4
3 1 4 2
2 6
3 1 7 5
1
3 3 6 4

Sample Output

5
4
6
9

Data Constraint

题目更正：vi值域小于等于1e9。

Solution

• 改修主席树？？

• 其实不需要，我们只需要倒过来建静态的主席树即可，因为插入删除都是在最左端。

• 为了节约一点点空间，我还打了个离散化。。

• 时间复杂度 $O(nlogn)$ 。

Code

#include<cstdio>
#include<algorithm>
#include<cctype>
using namespace std;
const int N=2e5+5;
struct data
{
	int v,l,r;
}f[N*38],c[N];
struct number
{
	int x,id;
}a[N<<1];
int tot,num,qx,len;
int rt[N<<1],rk[N<<1],pos[N];
inline int read()
{
    int X=0,w=0; char ch=0;
    while(!isdigit(ch)) w|=ch=='-',ch=getchar();
    while(isdigit(ch)) X=(X<<3)+(X<<1)+(ch^48),ch=getchar();
    return w?-X:X;
}
void write(int x)
{
	if(x>9) write(x/10);
	putchar(x%10+'0');
}
inline bool cmp(number x,number y)
{
	return x.x<y.x;
}
void make(int &v,int l,int r)
{
	v=++tot;
	if(l==r) return;
	int mid=l+r>>1;
	make(f[v].l,l,mid);
	make(f[v].r,mid+1,r);
}
void insert(int &x,int pre,int l,int r)
{
	f[x=++tot]=f[pre];
	f[x].v++;
	if(l==r) return;
	int mid=l+r>>1;
	if(qx<=mid) insert(f[x].l,f[pre].l,l,mid); else insert(f[x].r,f[pre].r,mid+1,r);
}
int find(int x,int pre,int l,int r,int k)
{
	if(l==r) return a[l].x;
	int mid=l+r>>1,v=f[f[x].l].v-f[f[pre].l].v;
	if(v>=k) return find(f[x].l,f[pre].l,l,mid,k);
	return find(f[x].r,f[pre].r,mid+1,r,k-v);
}
int main()
{
	freopen("pigeon.in","r",stdin);
	freopen("pigeon.out","w",stdout);
	int n=num=read(),m=read();
	for(int i=n;i;i--) a[a[i].id=i].x=read();
	for(int i=1;i<=m;i++)
	{
		c[i].v=read()-2;
		if(!c[i].v)
		{
			c[i].l=read();
			a[++num].x=c[i].l;
			a[num].id=num;
			pos[i]=num;
		}else
			if(c[i].v>0) c[i].l=read(),c[i].r=read(),c[i].v=read();
	}
	sort(a+1,a+1+num,cmp);
	for(int i=1;i<=num;i++) rk[a[i].id]=i;
	make(rt[0],1,num);
	for(int i=1;i<=n;i++)
	{
		len++;
		qx=rk[i];
		insert(rt[len],rt[len-1],1,num);
	}
	for(int i=1;i<=m;i++)
	{
		if(c[i].v<0)
		{
			rt[len--]=0;
		}else
		if(!c[i].v)
		{
			len++;
			qx=rk[pos[i]];
			insert(rt[len],rt[len-1],1,num);
		}else
		{
			int l=len-c[i].r+1,r=len-c[i].l+1,k=c[i].v;
			int ans=find(rt[r],rt[l-1],1,num,k);
			write(ans),putchar('\n');
		}
	}
	return 0;
}