在大学期间,经常需要租借教室。
大到院系举办活动,小到学习小组自习讨论,都需要向学校申请借教室。
教室的大小功能不同,借教室人的身份不同,借教室的手续也不一样。
面对海量租借教室的信息,我们自然希望编程解决这个问题。
我们需要处理接下来 n n n天的借教室信息,其中第 i i i天学校有 r i r_i ri个教室可供租借。
共有 m m m份订单,每份订单用三个正整数描述,分别为 d j , s j , t j d_j,s_j,t_j dj,sj,tj,表示某租借者需要从第 s j s_j sj天到第 t j t_j tj天租借教室(包括第 s j s_j sj天和第 t j t_j tj天),每天需要租借 d j d_j dj个教室。
我们假定,租借者对教室的大小、地点没有要求。
即对于每份订单,我们只需要每天提供 d j d_j dj个教室,而它们具体是哪些教室,每天是否是相同的教室则不用考虑。
借教室的原则是先到先得,也就是说我们要按照订单的先后顺序依次为每份订单分配教室。
如果在分配的过程中遇到一份订单无法完全满足,则需要停止教室的分配,通知当前申请人修改订单。
这里的无法满足指从第sj天到第tj天中有至少一天剩余的教室数量不足 d j d_j dj个。
现在我们需要知道,是否会有订单无法完全满足。
如果有,需要通知哪一个申请人修改订单。
输入格式
第一行包含两个正整数
n
,
m
n,m
n,m,表示天数和订单的数量。
第二行包含 n n n个正整数,其中第 i i i个数为 r i r_i ri,表示第i天可用于租借的教室数量。
接下来有 m m m行,每行包含三个正整数 d j , s j , t j d_j,s_j,t_j dj,sj,tj,表示租借的数量,租借开始、结束分别在第几天。
每行相邻的两个数之间均用一个空格隔开。
天数与订单均用从 1 1 1开始的整数编号。
输出格式
如果所有订单均可满足,则输出只有一行,包含一个整数
0
0
0。
否则(订单无法完全满足)输出两行,第一行输出一个负整数 − 1 -1 −1,第二行输出需要修改订单的申请人编号。
数据范围
1
≤
n
,
m
≤
106
,
0
≤
r
i
,
d
j
≤
109
,
1
≤
s
j
≤
t
j
≤
n
1≤n,m≤106,\\ 0≤r_i,d_j≤109,\\ 1≤s_j≤t_j≤n
1≤n,m≤106,0≤ri,dj≤109,1≤sj≤tj≤n
输入样例:
4 3
2 5 4 3
2 1 3
3 2 4
4 2 4
输出样例:
-1
2
- 思路(二分,差分) O((n+m)
由于随着订单数量的增加,每天可用教室的数量一定单调下降。因此我们可以二分出第一天出现负值的订单编号。
剩下的问题是如何快速求出经过若干订单后,每天所剩的教室数量。
每个订单的操作是 [ L i , R i ] [L_i,R_i] [Li,Ri]全部减去 d i d_i di。
因此我们可以用差分来加速处理过程。
时间复杂度分析
总共二分 O ( l o g m ) O(logm) O(logm)次,其中 m m m是订单数量。每次二分后使用差分求出每天最终教室数量,计算量是 O ( n + m ) O(n+m) O(n+m),因此总时间复杂度是 O ( ( n + m ) l o g m ) O((n+m)logm) O((n+m)logm)。
#include <bits/stdc++.h>
using namespace std;
typedef long long ll;
const int maxn = 1e6 + 10;
int n, m;
ll r[maxn];
int d[maxn], s[maxn], t[maxn];
ll a[maxn];
bool check(int x) {
for (int i = 1; i <= n; i++) {
a[i] = r[i];
}
for (int i = 1; i <= x; i++) {
a[s[i]] -= d[i];
a[t[i] + 1] += d[i];
}
ll sum = 0;
for (int i = 1; i <= n; i++) {
sum += a[i];
if (sum < 0) return 0;
}
return 1;
}
int ans;
int main() {
//freopen("1.in", "r", stdin);
scanf("%d%d", &n, &m);
for (int i = 1; i <= n; i++) {
scanf("%lld", &r[i]);
}
for (int i = n; i >= 1; i--) {
r[i] -= r[i - 1];
}
for (int i = 1; i <= m; i++) {
scanf("%d%d%d", &d[i], &s[i], &t[i]);
}
int el = 1, er = m;
while (el <= er) {
int mid = el + er >> 1;
if (check(mid))
el = mid + 1;
else {
er = mid - 1;
ans = mid;
}
}
if (check(ans)) {
puts("0");
} else {
puts("-1");
printf("%d\n", ans);
}
return 0;
}