hdu2871_Memory Control

http://acm.hdu.edu.cn/showproblem.php?pid=2871

由于做过hotel（poj3667），所以连数据规模都不用看就知道这题是线段树。

hotel的check-in对应此题的new，check-out对应free，build对应reset（一开始是打算用build来实现reset，后来注意到，直接用update把1到n全都设置为空即可，利用lazy update，否则重新build会tle）。

当然，由于Free操作跟hotel里的还不是完全一样，而且也多了个Get操作，本来还想修改query，后来发现，只需要维护多一个数组，按顺序记录每一个使用中的数据块的起始和结束位置即可。由于vector的插入和删除非常方便，所以就使用vector了~

特别注意：

1、在vector里查找某个unit在哪个block的时候，要用二分查找。。。TAT 我就是TLE死在这里。。。懂得把New的改为二分查找，居然忘了修改Free里面的，结果还是没拿到first blood，这道题到练习赛结束还是没人过。。

2、在Free操作里，二分查找使用upper_bound而不用lower_bound，因为如果遇到要Free的那个block刚好begin就是unit x，那么lower_bound的结果是那个block，而不是那个block的下一个位置，不需要再减1。

代码：

#include <cstdio>#include <algorithm>#include <vector>using namespace std;const int N = 50000 + 10;struct SegTree{int left, right;int maxval, lval, rval;int cover;int mid(){return (left + right) / 2;}int dis(){return right - left + 1;}void init(){maxval = lval = rval = cover ? 0 : dis();}}st[N*4];struct block{    int begin, end;};vector<block> v;bool cmp(const block &a, const block &b){    return a.begin < b.begin;}void build(int left, int right, int root){st[root].left = left;st[root].right = right;st[root].cover = 0;st[root].init();if (left == right)return;int mid = st[root].mid();build(left, mid, root*2);build(mid+1, right, root*2+1);}void update(int left, int right, int cov, int root){if (left <= st[root].left && st[root].right <= right){st[root].cover = cov;st[root].init();return;}if (st[root].cover != -1){st[root*2].cover = st[root*2+1].cover = st[root].cover;st[root*2].init();st[root*2+1].init();st[root].cover = -1;}int mid = st[root].mid();if (left <= mid)update(left, right, cov, root*2);if (right > mid)update(left, right, cov, root*2+1);st[root].maxval = max(st[root*2].rval + st[root*2+1].lval, max(st[root*2].maxval, st[root*2+1].maxval));st[root].lval = st[root*2].lval;st[root].rval = st[root*2+1].rval;if (st[root*2].lval == st[root*2].dis())st[root].lval += st[root*2+1].lval;if (st[root*2+1].rval == st[root*2+1].dis())st[root].rval += st[root*2].rval;}int query(int d, int root){if (st[root].maxval < d)return 0;if (st[root].left == st[root].right)return st[root].left;if (st[root].cover != -1){st[root*2].cover = st[root*2+1].cover = st[root].cover;st[root*2].init();st[root*2+1].init();st[root].cover = -1;}if (st[root*2].maxval >= d)return query(d, root*2);else if (st[root*2].rval + st[root*2+1].lval >= d)return st[root*2].right - st[root*2].rval + 1;else return query(d, root*2+1);}int main(){int n, m;while (scanf("%d%d", &n, &m) != EOF)    {        build(1, n, 1);        v.clear();        int x;        char s[6], ch;        while (m--)        {            scanf("%c", &ch);  //read '\n' in the end of last line            scanf("%s", s);            if (s[0] == 'R')            {//              build(1, n, 1);    //TLE!!!                update(1, n, 0, 1);                v.clear();                printf("Reset Now\n");            }            else if (s[0] == 'N')            {                scanf("%d", &x);                int a = query(x, 1);                if (a == 0)                    printf("Reject New\n");                else                {                    printf("New at %d\n", a);                    update(a, a + x - 1, 1, 1);                    //update v[]                    block y;                    y.begin = a, y.end = a + x - 1;                    vector<block>::iterator it;                    it = upper_bound(v.begin(), v.end(), y, cmp); //lower_bound() is all right.                    v.insert(it, y);//                  v.push_back(y);           //TLE!!!//                  sort(v.begin(), v.end(), cmp);                }            }            else if (s[0] == 'F')            {                scanf("%d", &x);                //find which block the x located                block y;                y.begin = x, y.end = x;                vector<block>::iterator it;                it = upper_bound(v.begin(), v.end(), y, cmp);  //lower_bound() may make the code uglier.                int tmp = it - v.begin() - 1;                if (tmp == - 1 || v[tmp].end < x)                    printf("Reject Free\n");                else                {                    printf("Free from %d to %d\n", v[tmp].begin, v[tmp].end);                    update(v[tmp].begin, v[tmp].end, 0, 1);                    v.erase(v.begin() + tmp);                }            }            else            {                scanf("%d", &x);                if (x > v.size())                    printf("Reject Get\n");                else printf("Get at %d\n", v[x-1].begin);            }        }        printf("\n");    }return 0;}