Hotel 【POJ - 3667】【线段树】

题目链接

  题意：让你求最先出现的能取长度为len的一段连续房间的最先的点，以及清扫房间，让从a~a+b-1的房间全都空出来。

  思路：

  一道线段树区间更新，区别就在于要输出的不再是这段连续区间，而是其第一位的序位，所以，我们构建线段树，先从其左儿子开始找，然后找中间，最后在找右儿子，返回就是了，我WA了几发，发现是线段树的更新错了，当时也是懵逼，竟然用这组错的数据连过好几组测试数据也是醉了。

建树的操作就不需要多说，我们来讲讲其他几个操作：

向下push_down（）：

void push_down(int rt, int l, int r)
{
    int mid=(l+r)>>1;
    if(lazy[rt]!=-1)
    {
        lazy[rt<<1]=lazy[rt];
        lazy[rt<<1|1]=lazy[rt];
        tree[rt<<1]=node(lazy[rt]*(mid-l+1), lazy[rt]*(mid-l+1), lazy[rt]*(mid-l+1));
        tree[rt<<1|1]=node(lazy[rt]*(r-mid), lazy[rt]*(r-mid), lazy[rt]*(r-mid));
        lazy[rt]=-1;
    }
}

  lazy标记一开始有-1，1，0这三种情况，分别表示{没有lazy、房间空出来了、房间使用了}向下递归就是了。

向上更新push_up（）：

void push_up(int rt, int l, int r)
{
    tree[rt].lc=tree[rt<<1].lc;
    tree[rt].rc=tree[rt<<1|1].rc;
    int mid=(l+r)>>1;
    if(mid-tree[rt].lc+1==l) tree[rt].lc+=tree[rt<<1|1].lc;
    if(r-tree[rt].rc==mid) tree[rt].rc+=tree[rt<<1].rc;
    tree[rt].mc=max(tree[rt].lc, tree[rt].rc);
    tree[rt].mc=max(tree[rt].mc, max(tree[rt<<1].mc, tree[rt<<1|1].mc));
    tree[rt].mc=max(tree[rt].mc, tree[rt<<1].rc+tree[rt<<1|1].lc);
}

  向上更新就是给予目前节点的赋值，我们考虑它的两个子节点，有左儿子、右儿子，对于父节点，左初始为左儿子的左，但要是是满的就可以再加上右儿子的左，同理，右儿子类似，然后在考虑最大值，这里考虑最大值的时候不要忘记中间节点，就是左儿子的右加上右儿子的左，其余应该不会忘记。

更新函数update（）：

void update(int rt, int l, int r, int ql, int qr, int val)
{
    if(ql<=l && qr>=r)
    {
        tree[rt]=node(val*(r-l+1), val*(r-l+1), val*(r-l+1));
        lazy[rt]=val;
        return;
    }
    push_down(rt, l, r);
    int mid=(l+r)>>1;
    if(ql>mid) update(rt<<1|1, mid+1, r, ql, qr, val);
    else if(qr<=mid) update(rt<<1, l, mid, ql, qr, val);
    else
    {
        update(rt<<1, l, mid, ql, mid, val);
        update(rt<<1|1, mid+1, r, mid+1, qr, val);
    }
    push_up(rt, l, r);
}

  更新函数还算好想，更新一段区间，给予其lazy标记，对于价值可以考虑为是否是空房，cal==1就是空房；val==0就是住了人的房间。

查找函数Query（）：
 

int Query(int rt, int l, int r, int val)
{
    if(tree[rt].mc<val) return 0;
    if(l==r) return l;
    int mid=(l+r)>>1;
    push_down(rt, l, r);
    if(tree[rt<<1].mc>=val) return Query(rt<<1, l, mid, val);
    else if(tree[rt<<1].rc+tree[rt<<1|1].lc>=val) return mid-tree[rt<<1].rc+1;
    else return Query(rt<<1|1, mid+1, r, val);
}

  查找函数就是找到第一个节点，我们遇到l==r就返回，那时候就是叶子节点了（重点），剩下的就是左中右的考虑了。

 

完整代码：

#include <iostream>
#include <cstdio>
#include <cmath>
#include <string>
#include <cstring>
#include <algorithm>
#include <limits>
#include <vector>
#include <stack>
#include <queue>
#include <set>
#include <map>
#define lowbit(x) ( x&(-x) )
#define pi 3.141592653589793
#define e 2.718281828459045
using namespace std;
typedef long long ll;
const int maxN=50005;
int N, M, lazy[maxN<<2];
struct node
{
    int lc, rc, mc;     //左、右、最大
    node(int a=0, int b=0, int c=0):lc(a), rc(b), mc(c) {}
}tree[maxN<<2];
void buildTree(int rt, int l, int r)
{
    lazy[rt]=-1;
    tree[rt]=node(r-l+1, r-l+1, r-l+1);
    if(l==r) return;
    int mid=(l+r)>>1;
    buildTree(rt<<1, l, mid);
    buildTree(rt<<1|1, mid+1, r);
}
void push_down(int rt, int l, int r)
{
    int mid=(l+r)>>1;
    if(lazy[rt]!=-1)
    {
        lazy[rt<<1]=lazy[rt];
        lazy[rt<<1|1]=lazy[rt];
        tree[rt<<1]=node(lazy[rt]*(mid-l+1), lazy[rt]*(mid-l+1), lazy[rt]*(mid-l+1));
        tree[rt<<1|1]=node(lazy[rt]*(r-mid), lazy[rt]*(r-mid), lazy[rt]*(r-mid));
        lazy[rt]=-1;
    }
}
void push_up(int rt, int l, int r)
{
    tree[rt].lc=tree[rt<<1].lc;
    tree[rt].rc=tree[rt<<1|1].rc;
    int mid=(l+r)>>1;
    if(mid-tree[rt].lc+1==l) tree[rt].lc+=tree[rt<<1|1].lc;
    if(r-tree[rt].rc==mid) tree[rt].rc+=tree[rt<<1].rc;
    tree[rt].mc=max(tree[rt].lc, tree[rt].rc);
    tree[rt].mc=max(tree[rt].mc, max(tree[rt<<1].mc, tree[rt<<1|1].mc));
    tree[rt].mc=max(tree[rt].mc, tree[rt<<1].rc+tree[rt<<1|1].lc);
}
void update(int rt, int l, int r, int ql, int qr, int val)
{
    if(ql<=l && qr>=r)
    {
        tree[rt]=node(val*(r-l+1), val*(r-l+1), val*(r-l+1));
        lazy[rt]=val;
        return;
    }
    push_down(rt, l, r);
    int mid=(l+r)>>1;
    if(ql>mid) update(rt<<1|1, mid+1, r, ql, qr, val);
    else if(qr<=mid) update(rt<<1, l, mid, ql, qr, val);
    else
    {
        update(rt<<1, l, mid, ql, mid, val);
        update(rt<<1|1, mid+1, r, mid+1, qr, val);
    }
    push_up(rt, l, r);
}
int Query(int rt, int l, int r, int val)
{
    if(tree[rt].mc<val) return 0;
    if(l==r) return l;
    int mid=(l+r)>>1;
    push_down(rt, l, r);
    if(tree[rt<<1].mc>=val) return Query(rt<<1, l, mid, val);
    else if(tree[rt<<1].rc+tree[rt<<1|1].lc>=val) return mid-tree[rt<<1].rc+1;
    else return Query(rt<<1|1, mid+1, r, val);
}
int main()
{
    while(scanf("%d%d", &N, &M)!=EOF)
    {
        buildTree(1, 1, N);
        while(M--)
        {
            int e1, e2, e3;
            scanf("%d", &e1);
            if(e1==1)
            {
                scanf("%d", &e2);
                if(tree[1].mc<e2) { printf("0\n"); continue; }
                int tmp=Query(1, 1, N, e2);
                printf("%d\n", tmp);
                update(1, 1, N, tmp, tmp+e2-1, 0);
            }
            else
            {
                scanf("%d%d", &e2, &e3);
                update(1, 1, N, e2, e2+e3-1, 1);
            }
        }
    }
    return 0;
}