poj 3667 (线段树区间合并)

题意：

一家旅馆共有N个房间，这N个房间是连成一排的，标号为1-N。

现在有很多旅客以组为单位前来入住，每组旅客的数量可以用DiDi来表示。

旅店的业务分为两种，入住和退房：

1、旅客入住时，第i组旅客需要根据他们的人数DiDi，给他们安排DiDi个连续的房间，并且房间号要尽可能的小。如果房间不够，则无法安排。

2、旅客退房时，第i组旅客的账单将包含两个参数XiXi和DiDi,你需要将房间号XiXi到Xi+Di−1Xi+Di−1之间的房间全部清空。

现在你需要帮助该旅馆处理M单业务。

旅馆最初是空的。

题解：

线段树维护区间紧靠左边，紧靠右边以及整体最大连续长度即可，更新则用一个 mark 数组记录更新状态向下推即可。

本来不想写这篇的，但是看到网上有的题解代码风格实在是看的想吐，故写一个自认为比较简洁的 AC 代码。

代码：

#include <bits/stdc++.h>
using namespace std;

typedef long long ll;
typedef unsigned long long ull;
#define int ll
#define PI acos(-1.0)
#define INF 0x3f3f3f3f3f3f3f3f
#define P pair<int, int>
#define fastio ios::sync_with_stdio(false), cin.tie(0)
const int mod = 998244353;
const int M = 1000000 + 10;
const int N = 50000 + 10;

int n, m;
int lma[N<<2], rma[N<<2], ma[N<<2], mark[N<<2];

void pushup(int l, int r, int rt)
{
    int mid = (l + r) >> 1;
    lma[rt] = lma[rt << 1];
    rma[rt] = rma[rt << 1 | 1];

    if(lma[rt << 1] == mid - l + 1) lma[rt] += lma[rt << 1 | 1];
    if(rma[rt << 1 | 1] == r - mid) rma[rt] += rma[rt << 1];
    ma[rt] = max(max(ma[rt << 1], ma[rt << 1 | 1]), rma[rt << 1] + lma[rt << 1 | 1]);
}

void pushdown(int l, int r, int rt)
{
    if(mark[rt] != -1) {
        int mid = (l + r) >> 1;
        ma[rt << 1] = lma[rt << 1] = rma[rt << 1] = mark[rt] ? mid - l + 1 : 0;
        ma[rt << 1 | 1] = lma[rt << 1 | 1] = rma[rt << 1 | 1] = mark[rt] ? r - mid : 0;
        mark[rt << 1] = mark[rt << 1 | 1] = mark[rt];
        mark[rt] = -1;
    }
}

void build(int l = 1, int r = n, int rt = 1)
{
    mark[rt] = -1;
    ma[rt] = lma[rt] = rma[rt] = r - l + 1;
    if(l == r) return ;

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

void update(int L, int R, int c, int l = 1, int r = n, int rt = 1)
{
    if(L <= l && r <= R) {
        mark[rt] = c;
        lma[rt] = rma[rt] = ma[rt] = c ? r - l + 1 : 0;
        return ;
    }
    pushdown(l, r, rt);

    int mid = (l + r) >> 1;
    if(L <= mid) update(L, R, c, l, mid, rt << 1);
    if(mid < R) update(L, R, c, mid + 1, r, rt << 1 | 1);

    pushup(l, r, rt);
}

int query(int d, int l = 1, int r = n, int rt = 1)
{
    if(l == r) return l;
    pushdown(l, r, rt);
    
    int mid = (l + r) >> 1;
    if(ma[rt << 1] >= d) return query(d, l, mid, rt << 1);
    if(rma[rt << 1] + lma[rt << 1 | 1] >= d) return mid - rma[rt << 1] + 1;
    if(ma[rt << 1 | 1] >= d) return query(d, mid + 1, r, rt << 1 | 1);
    return 0;
}

signed main()
{
    fastio;
    cin >> n >> m;
    build();
    for(int i = 1, op; i <= m; i ++) {
        cin >> op;
        if(op == 1) {
            int d, ans;
            cin >> d;
            ans = query(d);
            cout << ans << endl;
            if(ans) update(ans, ans + d - 1, 0);
        } else {
            int x, d;
            cin >> x >> d;
            update(x, x + d - 1, 1);
        }
    }

    return 0;
}

/*

  Rejoicing in hope, patient in tribulation.

*/