区间合并 内纯释放问题 hdu2871 Memory control

题目链接  hdu2871 Memory control

题目大意： 给n个单元内存，有 四个操作：

      1、New x     找一段长为x的空区间段填满。  （下面用0表示空单元，1表示非空单元。）

      2、Free x  释放包含x的区间段

      3、 Get x 找到第x个区间段

      4、将整个区间都置为空。

解题思路：可以说这题应该算是线段树之区间操作里面比较好的经典题了。要注意的地方很多。对vector容器的用法又了解了很多。

#include <cstdio>
#include <vector>
#include <iostream>
using namespace std;

#define  lz  2*u,l,mid
#define  rz  2*u+1,mid+1,r
const int maxn=50005;
int flag[4*maxn];

struct segment
{
    int lm; // 从区间左端点开始的连续0个数;
    int rm; // 以区间右端点结束的连续0个数;
    int sm; // 整个区间最长连续的0个数;
} tree[4*maxn];

struct node
{
    int s, d;
};
node tmp;
vector<node>vt;

void push_up(int u, int l, int r)
{
    int mid= (l+r)>>1;
    tree[u].lm= tree[u*2].lm;
    tree[u].rm= tree[u*2+1].rm;
    tree[u].sm= max( tree[u*2].sm, tree[u*2+1].sm );
    if( tree[u*2].lm == mid-l+1 ) tree[u].lm += tree[u*2+1].lm;
    if( tree[u*2+1].rm == r-mid ) tree[u].rm += tree[u*2].rm;
    int t= tree[u*2].rm + tree[u*2+1].lm;
    if( t > tree[u].sm )  tree[u].sm= t;
}

void push_down(int u, int l, int r)
{
    if(flag[u]>=0)
    {
        int mid=(l+r)>>1;
        flag[2*u]=flag[2*u+1]=flag[u];
        tree[2*u].lm= tree[2*u].rm= tree[2*u].sm= flag[u]?0:mid-l+1;
        tree[2*u+1].lm= tree[2*u+1].rm= tree[2*u+1].sm= flag[u]?0:r-mid;
        flag[u]=-1;
    }
}

void build(int u, int l, int r)
{
    flag[u]=-1;
    if(l==r)
    {
        tree[u].lm= tree[u].rm= tree[u].sm= 1;
        return;
    }
    int mid= (l+r)>>1;
    build(lz);
    build(rz);
    push_up(u, l, r);
}

void Update(int u, int l, int r, int tl, int tr, int c)
{
    if(tl<=l&&r<=tr)
    {
        tree[u].lm=tree[u].rm=tree[u].sm=c?0:r-l+1;
        flag[u]=c;
        return ;
    }
    push_down(u,l,r);
    int mid=(l+r)>>1;
    if(tr<=mid) Update(lz,tl,tr,c);
    else if(tl>mid) Update(rz,tl,tr,c);
    else
    {
        Update(lz,tl,mid,c);
        Update(rz,mid+1,tr,c);
    }
    push_up(u,l,r);
}

int  Query(int u, int l, int r, int p)
{
    if(tree[u].sm==p&&r-l+1==p)   ///不仅要tree[u].sm==p而且要整个区间都被0覆盖，否则会出错
    {
        return l;
    }
    push_down(u,l,r);
    int mid= (l+r)>>1, t;
    if(p<=tree[2*u].sm)  return Query(lz,p);
    else if(tree[2*u].rm+tree[2*u+1].lm>=p) return mid-tree[2*u].rm+1;
    else return Query(rz,p);
}

int find(int tp)    
{
    int l=0, r=vt.size()-1, mid, ans=-1;
    while(l<=r)
    {
        int mid=(l+r)>>1;
        if(vt[mid].s<=tp)
        {
            ans=mid;
            l=mid+1;
        }
        else r=mid-1;
    }
    return ans;
}

int main()
{
    int n, m;
    while(cin >> n >> m)
    {
        build(1,1,n);
        vt.clear();
        while(m--)
        {
            char ch[6];
            scanf("%s",ch);
            if(ch[0]=='R')
            {
                vt.clear();
                Update(1,1,n,1,n,0);
                puts("Reset Now");
                continue;
            }
            int op;
            scanf("%d",&op);
            if(ch[0]=='N')
            {
                if(tree[1].sm>=op)
                {
                    int st=Query(1,1,n,op);
                    tmp.s=st, tmp.d=st+op-1;
                    int id=find(tmp.s);
                    vt.insert(vt.begin()+id+1,tmp); 
                    
                    printf("New at %d\n",tmp.s);
                    Update(1,1,n,tmp.s,tmp.d,1);  ///!!!这里注意了，查询完了还要进行相应的更新
                }
                else
                    puts("Reject New");
            }
            else if(ch[0]=='G')
            {
                if(vt.size()>=op)
                    printf("Get at %d\n",vt[op-1].s);
                else puts("Reject Get");
            }
            else if(ch[0]=='F')
            {
                int id=find(op);
                if(id==-1||vt[id].d<op) puts("Reject Free");
                else
                {
                    Update(1,1,n,vt[id].s,vt[id].d,0);
                    printf("Free from %d to %d\n",vt[id].s,vt[id].d);
                    vt.erase(vt.begin()+id,vt.begin()+id+1); ///删除区间[s，d)内的元素，左闭右开
                }
            }
        }
        puts("");
    }
    return 0;
}