[2017纪中10-28]序列操作 贪心+线段树/二分

题面
显然每次操作贪心对最大的c[i]减一是正确的。但如果从大到小排序,减前面c[i]个并不能保证有序。例如4,3,2,2,2,1,c[i]=3,减得4,3,1,2,2,1。我们应该把第c[i]个数求出来,>c[i]的照常减,=c[i]的优先减靠右的即可。线段树维护。
听说还可以二分?
代码:

#include
#include
#include
#include
using namespace std;
const int maxn=1000010;
int n,m,a[1000010];
int read()
{
    int x=0;
    char ch=getchar();
    while(ch<'0'||ch>'9') ch=getchar();
    while(ch>='0'&&ch<='9') {x=x*10+ch-'0';ch=getchar();}
    return x;
}
struct tree
{
    int mi,mx,l,r,p;
    tree *ls,*rs;
    tree()
    {
        mi=mx=l=r=p=0;
        ls=rs=NULL;
    }
    void cal(int add){mi+=add;mx+=add;}
    void update()
    {
        mi=min(ls->mi,rs->mi);
        mx=max(ls->mx,rs->mx);
    }
    void pushdown()
    {
        if(p!=0)
        {
            ls->p+=p;rs->p+=p;
            ls->cal(p);rs->cal(p);
            p=0;
        }
    }
    void build(int lx,int rx)
    {
        l=lx;r=rx;
        if(l==r) {mi=mx=a[l];return;}
        int mid=(l+r)>>1;
        (ls=new tree)->build(lx,mid);
        (rs=new tree)->build(mid+1,rx);
        update();
    }
    void dec(int lx,int rx)
    {
        if(l==lx&&r==rx) {p--;cal(-1);return;}
        pushdown();
        int mid=(l+r)>>1;
        if(rx<=mid) ls->dec(lx,rx);
        else if(lx>mid) rs->dec(lx,rx);
        else ls->dec(lx,mid),rs->dec(mid+1,rx);
        update(); 
    }
    int query(int x)
    {
        if(l==r) {return mi;}
        pushdown();
        int mid=(l+r)>>1;
        if(x<=mid) return ls->query(x);
        else return rs->query(x);
    }
    int lbd(int c)
    {
        if(mxreturn l-1;
        if(mi>c||l==r) return r;
        pushdown();
        if(rs->mx>=c) return rs->lbd(c);
        else return ls->lbd(c);
    }
}*xtr;
int main()
{
    n=read();m=read();
    for(int i=1;i<=n;i++)
        a[i]=read();
    sort(a+1,a+n+1,greater<int>()); 
    (xtr=new tree)->build(1,n);
    int rnd=1;
    for(rnd=1;rnd<=m;rnd++)
    {
        int t=read(),o,r1,r2;
        if(t>n) {rnd--;break;}
        o=xtr->query(t);
        r1=xtr->lbd(o+1);
        r2=xtr->lbd(o);
        if(r1>=1) xtr->dec(1,r1);
        if(t-r1>=1) xtr->dec(r2-t+r1+1,r2);
        if(xtr->mi<0) {rnd--;break;}
    }
    printf("%d",min(m,rnd));
    return 0;
}

你可能感兴趣的:(数据结构,贪心算法,二分,线段树)