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