当前位置 博文首页 > ttoobne:poj 3667 (线段树区间合并)

    ttoobne:poj 3667 (线段树区间合并)

    作者:[db:作者] 时间:2021-08-30 10:35

    题意:

    一家旅馆共有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.
    
    */
    

    ?

    cs