Codeforces Round #668 (Div. 2) E

题意:
给出一个长度为 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;
}
  • 1
    点赞
  • 0
    收藏
    觉得还不错? 一键收藏
  • 0
    评论
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值