题意:
给出一个长度为 n 的数列 a ,规定当 a[ i ] == i 时,位置 i 可以被删除掉,后面位置会合并上来,现在需要回答 m 次询问,每次询问会问禁用掉后面 x 个数字和后面的 y 个数字后,最多可以删除掉多少个数字,每个询问都相互独立,举个例子比较好看:
思路:
首先询问给出的可以转化为给出l,r,然后对于一个区间的询问有一个很明显的贪心,就是每次把能去掉的最右边的去掉,这样就可以使得去掉的个数最大化。但是很明显多组询问就爆了,这时候就考虑怎么优化,其实对于一个数a[i]有三种情况:
1.a[i]>i,这时候其实他前面的数再怎么拿走,他也不可能被拿走了,所以就标为-1,即不可能去除节点。
2.a[i]=i,这时候就直接拿走即可。
3.a[i]<i,这时候需要前面拿走i-a[i]个才能让这个数被拿走。
这样我们可以维护一个l[i]数组,代表当询问的l小于等于l[i]时这个值的贡献为1,否则为0。
当这个数为上述的第一种情况时记为-1标记一下,代表不可能。
这个数组的l[i]其实就是前面所有l[j]中按权值的第num-(i-a[i])+1小,然后就可以用线段树求出第三种情况l[i]数组。如果是第二种情况就直接l[i]=i。
然后就可以把询问离线按照r来排序,这样按照询问的R顺序,依次把l[i]放入线段树中(记得这时候的线段树可以复用之前的那个线段树,因为维护的东西都是sum,记得build即可),然后就对于询问(L,R),R是不用考虑的,因为当前加入的全是R之前的,然后再把l[i]大于等于L的个数用线段树查出来即可。
#include<iostream>
#include<cstdio>
#include<algorithm>
using namespace std;
const int MAX_N=301000;
struct node{
int l,r,sum;
}a[MAX_N*4];
void update(int k){
a[k].sum=a[k<<1].sum+a[k<<1|1].sum;
}
void build(int k,int l,int r){
a[k].l=l;a[k].r=r;
if(l==r){
a[k].sum=0;
return;
}
int mid=(l+r)>>1;
build(k<<1,l,mid);
build(k<<1|1,mid+1,r);
update(k);
}
void change(int k,int x,int y){
if(a[k].l==a[k].r){
a[k].sum+=y;
return;
}
int mid=(a[k].l+a[k].r)>>1;
if(x<=mid)
change(k<<1,x,y);
else
change(k<<1|1,x,y);
update(k);
}
int query(int k,int p,int l,int r){//求第p小
if(l==r){
return l;
}
int mid=(l+r)>>1;
if(a[k<<1].sum>=p)
return query(k<<1,p,l,mid);
else
return query(k<<1|1,p-a[k<<1].sum,mid+1,r);
}
int ask(int k,int l,int r){
if(a[k].l>=l&&a[k].r<=r)
return a[k].sum;
int mid=(a[k].l+a[k].r)>>1;
int x=0;
if(r>mid)
x+=ask(k<<1|1,l,r);
if(l<=mid)
x+=ask(k<<1,l,r);
return x;
}
int l[MAX_N];
int num[MAX_N];
struct skt{
int l,r,num;
}b[MAX_N];
bool cmp(skt a,skt b){
return a.r<b.r;
}
int ans[MAX_N],A[MAX_N];
int main(void){
int n,q,i;
scanf("%d%d",&n,&q);
for(i=1;i<=n;i++)
scanf("%d",&A[i]);
int num=0;
build(1,1,n);
for(i=1;i<=n;i++){
if(A[i]>i){
l[i]=-1;
}
else if(A[i]==i){
l[i]=i;
num++;
change(1,l[i],1);
}
else{
int x=i-A[i];
int p=num-x+1;
if(p<=0)
l[i]=-1;
else{
l[i]=query(1,p,1,n);
change(1,l[i],1);
num++;
}
}
}
for(i=1;i<=q;i++){
scanf("%d%d",&b[i].l,&b[i].r);
b[i].l++;
b[i].r=n-b[i].r;
b[i].num=i;
}
sort(b+1,b+q+1,cmp);
int st=0;
build(1,1,n);
for(i=1;i<=q;i++){
while(st<b[i].r){
st++;
if(l[st]==-1)
continue;
change(1,l[st],1);
}
ans[b[i].num]=ask(1,b[i].l,n);
}
for(i=1;i<=q;i++)
printf("%d\n",ans[i]);
return 0;
}