1. 程式人生 > >JZOJ5933. 【NOIP2018模擬10.27】百鴿籠

JZOJ5933. 【NOIP2018模擬10.27】百鴿籠

題目

在JOU管理員群裡一共有n(n≤200000)個管理員,為了容納這些管理員,vfk準備了n個鴿籠。每個 鴿籠中都裝有一個咕咕能力值為vi的鴿子,每隻鴿子能力值不一定相同。每當RU開始或結束或 咕咕咕時,管理員們就會對這些鴿籠進行操作。操作包括三種:
1 vfk和管理員們取出最左端的鴿籠。
2 v vfk和管理員們在最左側新放入一個裝有咕咕能力值為v的鴿子的鴿籠。
3 l r k vfk想要知道從左到右第l個到第r個鴿籠中鴿子的第k小的咕咕能力值。 小∞ ⇔想要知道每個3操作(即詢問)的答案,請你來幫幫他。

題解

求區間第k小,很自然的想到主席樹,
可是這裡面有修改操作,
帶修主席樹?
顯然這個並不是聯賽考點,
注意觀察可以知道,它只是修改第一個位置,
如果將a反過來,那麼修改就是最後一個位置,
這樣就可以解決了修改的問題。
於是就變成了一個簡單的主席樹問題。

code

#include <cstdio>
#include <iostream>
#include <algorithm>
#include <cstring>
#include <cmath>
#define N 200003
#define P putchar
#define G getchar
using namespace std;
char ch;
void read(int &n)
{
	n=0;
	ch=G();
	while((ch<'0' || ch>'9') && ch!='-')ch=G();
	int w=1;
	if(ch=='-')w=-1,ch=G();
	while('0'<=ch && ch<='9')n=(n<<3)+(n<<1)+ch-'0',ch=G();
	n*=w;
}

void write(int x){if(x>9) write(x/10);P(x%10+'0');}

struct node
{
	int l,r,s;
}t[N*100];

int n,m,a[N],all,tot,root[N*2],v[N*2],k,V;
int cz[4][N];

void build(int pre,int x,int l,int r)
{
	if(l==r)
	{
		t[x].s=t[pre].s+1;
		return;
	}
	int m=(l+r)>>1;
	if(V<=m)
	{
		t[x].l=++tot;t[x].r=t[pre].r;
		build(t[pre].l,t[x].l,l,m);
	}
	else
	{
		t[x].l=t[pre].l;t[x].r=++tot;
		build(t[pre].r,t[x].r,m+1,r);
	}
	t[x].s=t[t[x].l].s+t[t[x].r].s;
}

void find(int x,int y,int l,int r)
{
	if(l==r)
	{
		V=l;
		return;
	}
	int v=t[t[y].l].s-t[t[x].l].s,m=(l+r)>>1;
	if(k<=v)find(t[x].l,t[y].l,l,m);
		else k=k-v,find(t[x].r,t[y].r,m+1,r);
}

int main()
{
	freopen("pigeon.in","r",stdin);
	freopen("pigeon.out","w",stdout);
	
	read(n);read(m);all=n;
	for(int i=1;i<=n;i++)read(a[i]),v[i]=a[i];
	for(int i=1;i<=m;i++)
	{
		read(cz[0][i]);
		if(cz[0][i]==2)read(cz[1][i]),v[++all]=cz[1][i];else
		if(cz[0][i]==3)read(cz[1][i]),read(cz[2][i]),read(cz[3][i]);
	}
	sort(v+1,v+1+all);
	all=unique(v+1,v+1+all)-v-1;
	for(int i=1;i<=n;i++)
		root[i]=++tot,V=lower_bound(v+1,v+1+all,a[n-i+1])-v,
			build(root[i-1],root[i],1,all);
	
	for(int i=1;i<=m;i++)
	{
		if(cz[0][i]==1)n--;else
		if(cz[0][i]==2)root[++n]=++tot,V=lower_bound(v+1,v+1+all,cz[1][i])-v,
			build(root[n-1],root[n],1,all);
		else
		{
			k=cz[3][i];
			find(root[n-cz[2][i]],root[n-cz[1][i]+1],1,all);
			write(v[V]);P('\n');
		}
	}
	
	return 0;
}