HDU 2871 Memory Control

一共4种操作

其中用线段树 区间合并，来维护连续空的长度，和找出那个位置。其他用vector维护即可

#include<cstring>
#include<cstdio>
#include<algorithm>
#include<cmath>
#include <iostream>
#include<vector>
#define lson i<<1
#define rson i<<1|1
#define N 50050
using namespace std;
struct node
{
    int l,r;
};
bool operator<(node a,node b)
{
    return a.l<b.l;
}
vector<node>e;
int lsum[N*],rsum[N*],msum[N*];
int flag[N*];
inline int max(int x,int y)
{
    return x>y?x:y;
}
void pushup(int i,int l,int r)
{
    int mid=(l+r)>>;
    lsum[i]=lsum[lson];
    rsum[i]=rsum[rson];
    if(lsum[i]==mid-l+)
        lsum[i]+=lsum[rson];
    if(rsum[i]==r-mid)
        rsum[i]+=rsum[lson];
    msum[i]=max(msum[lson],msum[rson]);
    msum[i]=max(msum[i],rsum[lson]+lsum[rson]);
}
void pushdown(int i,int l,int r)
{
    if(flag[i]!=-)
    {
        int mid=(l+r)>>;
        flag[lson]=flag[rson]=flag[i];
        lsum[lson]=rsum[lson]=msum[lson]=(mid-l+)*flag[i];
        lsum[rson]=rsum[rson]=msum[rson]=(r-mid)*flag[i];
        flag[i]=-;
    }
}
void build(int l,int r,int i)
{
    flag[i]=-;
    lsum[i]=rsum[i]=msum[i]=r-l+;
    if(l==r)
        return ;
    int mid=(l+r)>>;
    build(l,mid,lson);
    build(mid+,r,rson);
}
void update(int l,int r,int pl,int pr,int va,int i)
{
    if(l>=pl&&r<=pr)
    {
        flag[i]=va;
        rsum[i]=lsum[i]=msum[i]=(r-l+)*va;
        return ;
    }
    pushdown(i,l,r);
    int mid=(l+r)>>;
    if(pl<=mid)update(l,mid,pl,pr,va,lson);
    if(pr>mid)update(mid+,r,pl,pr,va,rson);
    pushup(i,l,r);
}
int query(int l,int r,int va,int i)
{
    if(msum[i]==r-l+)
        return l;
    pushdown(i,l,r);
    int mid=(l+r)>>;
    if(msum[lson]>=va)
        return query(l,mid,va,lson);
    else if(rsum[lson]+lsum[rson]>=va)
        return mid-rsum[lson]+;
    else
        return query(mid+,r,va,rson);
}
int main() {
    int n,m;
    char s[];
    while(scanf("%d%d",&n,&m)!=EOF)
    {
        build(,n,);
        e.clear();
        while(m--)
        {
            int x;
            vector<node>::iterator it;
            node d;
            scanf(" %s",s);
            if(s[]=='N')
            {
                scanf("%d",&x);
                if(msum[]<x)
                    puts("Reject New");
                else
                {
                    int tmp=query(,n,x,);
                    printf("New at %d\n",tmp);
                    d.l=tmp;d.r=tmp+x-;
                    update(,n,tmp,tmp+x-,,);
                    it=upper_bound(e.begin(),e.end(),d);
                    e.insert(it,d);
                }
            }
            else if(s[]=='F')
            {
                scanf("%d",&x);
                d.l=x;d.r=x;
                it=upper_bound(e.begin(),e.end(),d);
                int tmp=it-e.begin()-;
                if(tmp<||e[tmp].l>x||e[tmp].r<x)
                     printf("Reject Free\n");
                else
                {
                    printf("Free from %d to %d\n",e[tmp].l,e[tmp].r);
                    update(,n,e[tmp].l,e[tmp].r,,);
                    e.erase(e.begin()+tmp);
                }
            }
            else if(s[]=='G')
            {
                scanf("%d",&x);
                if(e.size()<x)
                    puts("Reject Get");
                else
                    printf("Get at %d\n",e[x-].l);
            }
            else
            {
                update(,n,,n,,);
                e.clear();
                puts("Reset Now");
            }
        }
        printf("\n");
    }
    return ;
}