HDU 2871 Memory Control

题目链接：http://acm.hdu.edu.cn/showproblem.php?pid=2871

题意：有内存大小为n，有4种操作：

1.New x : 分配大小为x的内存，尽量靠左，输出左边界

2.Free x : 释放包含x位置的内存块，输出左右边界

3.Get  x : 返回第x个内存块的左边界，内存块要从左到右排序

4 Reset : 恢复初始化的状态

思路：线段树区间合并，和hotel那一题完全一样，主要是几个操作的处理十分恶心，用vector二分的方法找出要释放的内存块，插入时二分寻找合适的位置避免排序超时(我对vector的操作不是很熟悉参考了别人的代码)

#include <iostream>
#include <cstdio>
#include <algorithm>
#include <cstring>
#include <vector>
#define maxn 50030
using namespace std;

struct Tree
{
    int l,r;
    int lsum,rsum,tsum;
}tree[maxn*3];

int lazy[maxn*3];

struct Memory
{
    int l,r;
};

vector <Memory> s;

bool cmp(Memory u,Memory v)
{
    return u.l<v.l;
}

void Pushup(int root)
{
    int ll=tree[root<<1].r-tree[root<<1].l+1;
    int rr=tree[root<<1|1].r-tree[root<<1|1].l+1;
    tree[root].lsum=tree[root<<1].lsum;
    tree[root].rsum=tree[root<<1|1].rsum;

    if (tree[root].lsum==ll) tree[root].lsum+=tree[root<<1|1].lsum;
    if (tree[root].rsum==rr) tree[root].rsum+=tree[root<<1].rsum;

    int tem=max(tree[root<<1].tsum,tree[root<<1|1].tsum);
    tree[root].tsum=max(tem,tree[root<<1].rsum+tree[root<<1|1].lsum);
}

void Pushdown(int root)
{
    if (lazy[root]!=-1)
    {
        lazy[root<<1]=lazy[root];
        lazy[root<<1|1]=lazy[root];
        if (lazy[root]==1)
        {
            tree[root<<1].lsum=tree[root<<1].rsum=tree[root<<1].tsum=0;
            tree[root<<1|1].lsum=tree[root<<1|1].rsum=tree[root<<1|1].tsum=0;
        }
        else
        {
            tree[root<<1].lsum=tree[root<<1].rsum=tree[root<<1].tsum=tree[root<<1].r-tree[root<<1].l+1;
            tree[root<<1|1].lsum=tree[root<<1|1].rsum=tree[root<<1|1].tsum=tree[root<<1|1].r-tree[root<<1|1].l+1;
        }
        lazy[root]=-1;
    }
}

void build(int root,int l,int r)
{
    tree[root].l=l;
    tree[root].r=r;
    tree[root].lsum=tree[root].rsum=tree[root].tsum=r-l+1;
    if (l==r) return;

    int mid=(l+r)>>1;
    build(root<<1,l,mid);
    build(root<<1|1,mid+1,r);
    Pushup(root);
}

void update(int root,int l,int r,int val)
{
    if (tree[root].l>=l && tree[root].r<=r)
    {
        lazy[root]=val;
        if (val==1)
        tree[root].lsum=tree[root].rsum=tree[root].tsum=0;
        else tree[root].lsum=tree[root].rsum=tree[root].tsum=tree[root].r-tree[root].l+1;
        return;
    }
    Pushdown(root);

    int mid=(tree[root].l+tree[root].r)>>1;
    if (l<=mid) update(root<<1,l,r,val);
    if (r>mid) update(root<<1|1,l,r,val);
    Pushup(root);
}

int que(int root,int w)
{
    if (tree[root].l==tree[root].r) return tree[root].l;

    Pushdown(root);
    int mid=(tree[root].l+tree[root].r)>>1;
    if (w<=tree[root<<1].tsum) que(root<<1,w);
    else if (w<=tree[root<<1].rsum+tree[root<<1|1].lsum) return mid-tree[root<<1].rsum+1;
    else return que(root<<1|1,w);

}

int main()
{
    int n,m;
    while (scanf("%d%d",&n,&m)!=EOF)
    {
        int cnt=0;
        memset(lazy,-1,sizeof(lazy));
        s.clear();
        build(1,1,n);
        for (int i=0;i<m;i++)
        {
            char ch[10];
            scanf("%s",&ch);
            if (ch[0]=='N')
            {
                int w,tem;
                Memory p;
                scanf("%d",&w);
                if (tree[1].tsum>=w)
                {
                    tem=que(1,w);
                    update(1,tem,tem+w-1,1);
                    p.l=tem;
                    p.r=tem+w-1;

                    vector<Memory>::iterator it;
                    it=upper_bound(s.begin(),s.end(),p,cmp);
                    s.insert(it,p);
                    printf("New at %d\n",tem);
                }
                else
                {
                    printf("Reject New\n");
                }
            }
            else if (ch[0]=='F')
            {
                Memory p;
                vector<Memory>::iterator it;
                int num,tem;
                scanf("%d",&num);
                p.l=num;
                p.r=num;
                it=upper_bound(s.begin(),s.end(),p,cmp);
                tem=it-s.begin()-1;
                if (tem==-1 || s[tem].r<num)
                {
                    printf("Reject Free\n");
                }
                else
                {
                    update(1,s[tem].l,s[tem].r,0);
                    printf("Free from %d to %d\n",s[tem].l,s[tem].r);
                    s.erase(s.begin()+tem);
                }
            }
            else if (ch[0]=='G')
            {
                int num;
                scanf("%d",&num);
                if (num<=s.size())
                printf("Get at %d\n",s[num-1].l);
                else printf("Reject Get\n");
            }
            else if (ch[0]=='R')
            {
                update(1,1,n,0);
                s.clear();
                printf("Reset Now\n");
            }
        }
        printf("\n");
    }
}