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
-
改修主席树??
-
其实不需要,我们只需要倒过来建静态的主席树即可,因为插入删除都是在最左端。
-
为了节约一点点空间,我还打了个离散化。。
-
时间复杂度 。
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;
}