NOIP模拟 序列操作【线段树】

题目大意：

一开始有n个非负整数a[i]，接下来会有m次操作，第i次操作给出一个数c[i]，要求选出c[i]个大于0的数并将它们减一。
问最多能进行多少轮操作后无法操作（即没有c[i]个大于0的数）。

解题思路：

首先可以发现减1的c[i]个数一定是最大的c[i]个数，所以当第c[i]大的数等于0时就无法再操作了，那么我们将a[i]从大到小排序。

注意到由于只减一，所以操作后除最小的一段数其余仍是单调递减的，如
333222211将前5个数减一后为：2221122211，只有2所在一段减一后不是单调的，但我们可以换成把最右边两个2减一就都单调了：222221111。

所以用线段树维护区间减即可。

#include<iostream>
#include<cstdio>
#include<cstring>
#include<string>
#include<algorithm>
#include<cmath>
#include<vector>
#include<queue>
#define ll long long
using namespace std;

int getint()
{
    int i=0,f=1;char c;
    for(c=getchar();(c<'0'||c>'9')&&c!='-';c=getchar());
    if(c=='-')f=-1,c=getchar();
    for(;c>='0'&&c<='9';c=getchar())i=(i<<3)+(i<<1)+c-'0';
    return i*f;
}

const int N=1000005;
int n,m;
int a[N],tr[N<<2],tag[N<<2];

bool cmp(const int &a,const int &b)
{
    return a>b;
}

void build(int k,int l,int r)
{
    if(l==r)
    {
        tr[k]=a[l];
        return;
    }
    int mid=l+r>>1;
    build(k<<1,l,mid),build(k<<1|1,mid+1,r);
    tr[k]=max(tr[k<<1],tr[k<<1|1]);
}

void pushdown(int k)
{
    tr[k<<1]+=tag[k],tag[k<<1]+=tag[k];
    tr[k<<1|1]+=tag[k],tag[k<<1|1]+=tag[k];
    tag[k]=0;
}

int query(int k,int l,int r,int p)
{
    if(l==r)return tr[k];
    if(tag[k])pushdown(k);
    int mid=l+r>>1;
    if(p<=mid)return query(k<<1,l,mid,p);
    else return query(k<<1|1,mid+1,r,p);
}

int find(int k,int l,int r,int v)
{
    if(l==r)return l;
    if(tag[k])pushdown(k);
    int mid=l+r>>1;
    if(tr[k<<1|1]>=v)return find(k<<1|1,mid+1,r,v);
    else find(k<<1,l,mid,v);
}

void modify(int k,int l,int r,int x,int y)
{
    if(x>y)return;
    if(x<=l&&r<=y)
    {
        tr[k]--,tag[k]--;
        return;
    }
    if(tag[k])pushdown(k);
    int mid=l+r>>1;
    if(y<=mid)modify(k<<1,l,mid,x,y);
    else if(x>mid)modify(k<<1|1,mid+1,r,x,y);
    else modify(k<<1,l,mid,x,mid),modify(k<<1|1,mid+1,r,mid+1,y);
    tr[k]=max(tr[k<<1],tr[k<<1|1]);
}

int main()
{
    //freopen("sequence.in","r",stdin);
    //freopen("sequence.out","w",stdout);
    n=getint(),m=getint();
    for(int i=1;i<=n;i++)a[i]=getint();
    sort(a+1,a+n+1,cmp);
    build(1,1,n);
    for(int i=1;i<=m;i++)
    {
        int c=getint(),x=query(1,1,n,c);
        if(x<=0)
        {
            cout<<i-1<<'\n';
            return 0;
        }
        int l=tr[1]<=x?0:find(1,1,n,x+1),r=find(1,1,n,x);
        modify(1,1,n,1,l);
        l=max(l,r-c+l)+1;
        modify(1,1,n,l,r);
    }
    cout<<m<<'\n';
    return 0;
}