Parking Lot题解

一道线段树区间合并，一开始本小白并不会写，看了大佬的题解后才写出来。
传送门（Codeforces 46D）

思路：这道题的精髓在于最开始建树时，多加一个b和f，这样我们就很好停车了（大家可以自己仔细思考一下），当我们这样建好树之后就是一个很经典的区间合并了。

#include <iostream>
#include <cstdio>
#include <cstring>
#include <string>
#include <algorithm>
#include <vector>
#include <queue>
#include <cmath>
#include <stack>
#define ll long long
#define INF 100000000
using namespace std;

const int N = 1e5 + 5;
const int M = 2005;
const int inf = 0x3f3f3f3f;
const int mod = 10007;

int n,f,b,sum[N<<2],lsum[N<<2],rsum[N<<2],add[N<<2],len[N],p[N]; 
// len用于储存每辆车的长度，p用于储存每辆车的起点，add用于标记是否停车

void pushup(int rt, int m){ // 经典区间合并，m为区间长度
    lsum[rt] = lsum[rt << 1];
    if(lsum[rt] == m - (m >> 1))
        lsum[rt] += lsum[rt << 1 | 1];
    rsum[rt] = rsum[rt << 1 | 1];
    if(rsum[rt] == (m >> 1))
        rsum[rt] += rsum[rt << 1];
    sum[rt] = max(max(sum[rt << 1], sum[rt << 1 | 1]), rsum[rt << 1] + lsum[rt << 1 | 1]);
    return ;
}

void pushdown(int rt, int m){ // 是否停车的标记下推，m为区间长度
    if(add[rt] != -1) {
        add[rt << 1] = add[rt << 1 | 1] = add[rt];
        sum[rt << 1] = lsum[rt << 1] = rsum[rt << 1] = add[rt] ? 0 : m - (m >> 1);
        sum[rt << 1 | 1] = lsum[rt << 1 | 1] = rsum[rt << 1 | 1] = add[rt] ? 0 : (m >> 1);
        add[rt] = -1;
    }
    return ;
}

void build(int l, int r, int rt) { //建树
    sum[rt] = lsum[rt] = rsum[rt] = r - l + 1; //初始化
    add[rt] = -1;
    if (l == r) {
        return;
    }
    int m = (l + r) >> 1;
    build(l, m, rt << 1);
    build(m + 1, r, rt << 1 | 1);
}

void Add(int x, int l, int r, int L, int R, int rt){//x == 1 停车，x == 0 取车
    if(l >= L && r <= R){
        sum[rt] = lsum[rt] = rsum[rt] = x ? 0 : r - l + 1;
        add[rt] = x;
        return ;
    }
    int m = (l + r) >> 1;
    pushdown(rt, r - l + 1);
    if(L <= m)
        Add(x, l, m, L, R, rt << 1);
    if(m < R)
        Add(x, m+1, r, L, R, rt << 1 | 1);
    pushup(rt, r - l + 1);
}

int query(int l, int r, int x, int rt){
    if(l == r){
        return l;
    }
    pushdown(rt, r - l + 1);
    int m = (l + r) >> 1;
    //仔细思考一下这里的顺序，必须是从左子树开始（如果从右子树就是反向停车啦）
    if(sum[rt << 1] >= x)
        return query(l, m, x, rt << 1);
    else if(rsum[rt << 1] + lsum[rt << 1 | 1] >= x)
        return m - rsum[rt << 1] + 1;
    else
        return query(m + 1, r, x, rt << 1 | 1);
}

int main() {
    scanf("%d%d%d",&n,&b,&f);
    build(1, b+n+f, 1); //多加一个b和f
    int q;
    scanf("%d",&q);
    for(int i=1; i<=q; i++){
        int x,y;
        scanf("%d%d",&x,&y);
        if(x == 1){
            if(b + y + f > sum[1]){ // 如果剩下的长度不够你停这辆车输出-1
                printf("-1\n");
                continue;
            }
            int pos = query(1, b+n+f, b+y+f, 1);
            printf("%d\n",pos-1);
            Add(1, 1, b+f+n, pos+b, pos+y+b-1, 1);
            p[i] = pos; // 记录此辆车的信息
            len[i] = y;
        }
        else{
            Add(0, 1, b+f+n, b+p[y], b+p[y]+len[y]-1, 1);
        }
    }
    return 0;
}