hdu 2871 Memory Control

线段树

题意：1到n的线段，m个操作

Reset 将所有内存清空

New x 申请一段长度为x的最左的未被占用的区间，返回它的位置，并占用它。找不到输出Reject New

Free x 找到x单元内存所在的连续的单元块，返回这个单元块的左右端点，并且清空这个单元块，如果这个单元x根本没被占用，则输出Reject Free

Get x 找到第x个单元块（而不是第x单元），返回这个单元块的起始位置即左端点

这题还是和 poj 3667 Hotel 是一样的，查询一个最左的合法区间，修改一段区间

这题使用vector按顺序来保存单元块，方便查找（可以使用二分）

而线段树主要的query和updata函数是一样的

#include <cstdio>
#include <cstring>
#include <vector>
using namespace std;
#define lch(i) ((i)<<1)
#define rch(i) ((i)<<1|1)
#define max(a,b) ((a)>(b)?(a):(b))
#define min(a,b) ((a)<(b)?(a):(b))
#define N 50010

struct block{
    int l,r;
};
struct node{
    int l,r;
    int mark;
    int tlen,llen,rlen;
    int mid(){
        return (l+r)>>1;
    }
    int cal_len(){
        return r-l+1;
    }
    void updata_len(){
        tlen=llen=rlen= (mark ? 0 : cal_len());
    }
};
typedef struct block Block;
typedef struct node node;
vector <Block> B;
node t[4*N];

void build(int l ,int r ,int rt)
{
    t[rt].l = l; t[rt].r = r; 
    t[rt].mark = 0;
    t[rt].updata_len();
    if(l == r) return ;
    int mid = t[rt].mid();
    int ll = lch(rt);
    int rr = rch(rt);
    build(l , mid , ll);
    build(mid+1 , r , rr);
    return ;
}

void updata(int l , int r , int val , int rt)
{
    if(t[rt].l == l && t[rt].r == r) 
    {
        t[rt].mark = val;
        t[rt].updata_len();
        return ;
    }
    int mid = t[rt].mid();
    int ll = lch(rt);
    int rr = rch(rt);
    
    if(t[rt].mark != -1)
    {
        t[ll].mark = t[rr].mark = t[rt].mark;
        t[rt].mark = -1;
        t[ll].updata_len();
        t[rr].updata_len();
    }

    if(l > mid)        updata(l , r , val , rr);
    else if(r <= mid)  updata(l , r , val , ll);
    else
    {
        updata(l , mid , val , ll);
        updata(mid+1 , r , val , rr);
    }

    t[rt].tlen = max( max(t[ll].tlen , t[rr].tlen)  ,  t[ll].rlen+t[rr].llen);
    t[rt].llen = t[ll].llen;  t[rt].rlen = t[rr].rlen;
    if(t[ll].tlen == t[ll].cal_len()) t[rt].llen += t[rr].llen;
    if(t[rr].tlen == t[rr].cal_len()) t[rt].rlen += t[ll].rlen;
    return ;
}

int query(int w , int rt)
{
    if(t[rt].l == t[rt].r)
    {
        if(w == 1) return t[rt].l;
        else       return 0;
    }

    int mid = t[rt].mid();
    int ll = lch(rt);
    int rr = rch(rt);
    
    if(t[rt].mark != -1)
    {
        t[ll].mark = t[rr].mark = t[rt].mark;
        t[rt].mark = -1;
        t[ll].updata_len();
        t[rr].updata_len();
    }

    if(t[ll].tlen >= w) 
        return query(w , ll);
    else if(t[ll].rlen + t[rr].llen >= w)
        return t[ll].r - t[ll].rlen + 1;
    else if(t[rr].tlen >= w)
        return query(w , rr);
    else
        return 0;
}

int binarysearch(int key)
{
    int low = 0;
    int high = B.size() - 1;
    while(low <= high)
    {
        int mid = (low + high) >> 1;
        if(B[mid].l <= key)
            low = mid + 1;
        else
            high = mid - 1;
    }
    return low;
    //此二分查找最终返回low,二分查找要写好，否则很难查错
    //B[low]代表的块是大于pos的，B[low-1]小于pos
    //对于添加一个新的区间，是在low这里添加，原来的low以及后面的区间往后移动
    //对于删除一个区间，是low-1，low以及后面的区间往前移动
}

int main()
{
    int n,m;
    while(scanf("%d%d",&n,&m)!=EOF)
    {
        B.clear();
        build(1,n,1);
        while(m--)
        {
            char op[10];
            int x,pos,index;
            Block tmp;
            scanf("%s",op);
            if(!strcmp(op,"Reset"))
            {
                updata(1,n,0,1);
                B.clear();
                puts("Reset Now");
                continue;
            }

            scanf("%d",&x);
            if(!strcmp(op,"New"))
            {
                pos = query(x,1);
                if(!pos) 
                {
                    puts("Reject New");
                    continue;
                }
                printf("New at %d\n",pos);
                tmp.l = pos;  tmp.r = pos + x -1;
                index = binarysearch(pos);
                B.insert(B.begin()+index , tmp);
                updata(tmp.l , tmp.r , 1 , 1);
            }
            else if(!strcmp(op,"Free"))
            {
                index = binarysearch(x) - 1;
                if(index == -1 || B[index].r < x)
                {
                    puts("Reject Free");
                    continue;
                }
                printf("Free from %d to %d\n",B[index].l , B[index].r);
                updata(B[index].l , B[index].r , 0 , 1);
                B.erase(B.begin()+index , B.begin()+index+1);
            }
            else  //Get
            {
                if(--x >= B.size()) 
                {
                    puts("Reject Get");
                    continue;
                }
                printf("Get at %d\n",B[x].l);
            }
        }
        printf("\n");
    }
    return 0;
}