AcWing 503. 借教室（超时原因，二分详解）

在大学期间，经常需要租借教室。

大到院系举办活动，小到学习小组自习讨论，都需要向学校申请借教室。

教室的大小功能不同，借教室人的身份不同，借教室的手续也不一样。

面对海量租借教室的信息，我们自然希望编程解决这个问题。

我们需要处理接下来 n 天的借教室信息，其中第 i 天学校有 ri 个教室可供租借。

共有 m 份订单，每份订单用三个正整数描述，分别为 dj,sj,tj，表示某租借者需要从第 sj 天到第 tj 天租借教室（包括第 sj 天和第 tj 天），每天需要租借 dj 个教室。

我们假定，租借者对教室的大小、地点没有要求。

即对于每份订单，我们只需要每天提供 dj 个教室，而它们具体是哪些教室，每天是否是相同的教室则不用考虑。

借教室的原则是先到先得，也就是说我们要按照订单的先后顺序依次为每份订单分配教室。

如果在分配的过程中遇到一份订单无法完全满足，则需要停止教室的分配，通知当前申请人修改订单。

这里的无法满足指从第 sj 天到第 tj 天中有至少一天剩余的教室数量不足 dj 个。

现在我们需要知道，是否会有订单无法完全满足。

如果有，需要通知哪一个申请人修改订单。

输入格式
第一行包含两个正整数 n,m，表示天数和订单的数量。

第二行包含 n 个正整数，其中第 i 个数为 ri，表示第 i 天可用于租借的教室数量。

接下来有 m 行，每行包含三个正整数 dj,sj,tj，表示租借的数量，租借开始、结束分别在第几天。

每行相邻的两个数之间均用一个空格隔开。

天数与订单均用从 1 开始的整数编号。

输出格式
如果所有订单均可满足，则输出只有一行，包含一个整数 0。

否则（订单无法完全满足）输出两行，第一行输出一个负整数 −1，第二行输出需要修改订单的申请人编号。

数据范围
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

超时代码：没有使用二分，时间复杂度是 n2 级别的

#include<iostream>
#include<string>
#include<algorithm>
#include<bits/stdc++.h>
#include<stack>
#include<set>
#include<vector>
#include<map>
#include<queue>
#include<deque>
#include<cctype>
#include<unordered_set>
#include<unordered_map>
#include<fstream>
#include<cstring>
using namespace std;
const int N = 1000010;
long long int a[N] = {0};
long long int b[N] = {0};
int main() {
	int n, m;
	cin >> n >> m;
	for (int i = 1; i <= n; i++) {
		cin >> a[i];
		b[i] = a[i] - a[i - 1];
	}

	for (int k = 1; k <= m; k++) {
		long long int d;
		int s, t;
		cin >> d >> s >> t;
		for (int i = s; i <= t; i++) {
			if (a[i] < d) {
				cout << "-1" << endl << k;
				return 0;
			}
		}
		b[s] -= d;
		b[t + 1] += d;
		for (int i = 1; i <= n; i++) {
			b[i] += b[i - 1];
			a[i] = b[i];
		}
		for (int i = 1; i <= n; i++) {
			b[i] = a[i] - a[i - 1];
		}

	}
	cout << "0";
	return 0;
}

使用二分， AC 代码：

详解如下：

从上面的代码，我们得知，如果一次次的比较，会超时，那么我们想到能不能使用二分查找呢，随着借的人的增加，教师的余额是线性减少的，所以，满足二分法的条件

输入数组存储，从第一天开始，二分查找，找到第一次不满足教师条件的天，如果此天不满足，说明此天是第一天不满足的，输出即可，如果此天满足，说明都满足，输出 0。

#include<iostream>
#include<string>
#include<algorithm>
#include<bits/stdc++.h>
#include<stack>
#include<set>
#include<vector>
#include<map>
#include<queue>
#include<deque>
#include<cctype>
#include<unordered_set>
#include<unordered_map>
#include<fstream>
#include<cstring>
using namespace std;
const int N = 1000010;
long long int a[N] = {0};
long long int b[N] = {0};
long long int bb[N] = {0};
int n, m;
int d[N], s[N], t[N];
bool check(int k) {// k 指的是 第 k 天
	for (int i = 1; i <= n; i++) {// 每次，重置差分数组
		bb[i] = b[i];
	} 
	for (int i = 1; i <= k; i++) {// 计算 k 天以后的查分数组
		bb[s[i]] -= d[i];
		bb[t[i] + 1] += d[i];
	}
	long long int sum = 0;
	for (int i = 1; i <= n; i++) {// 差粉数组第 i 次累加的和，指的是原数组的 a[i]
		sum += bb[i];// 如果存在某一天的余额小于 0，说明不满足
		if (sum < 0) {
			return true;
		}
	}
	return false;
}
int main() {
	cin >> n >> m;
	for (int i = 1; i <= n; i++) {
		cin >> a[i];
		b[i] = a[i] - a[i - 1];
	}
	for (int i = 1; i <= m; i++) {
		cin >> d[i] >> s[i] >> t[i];
	}
	int l = 1, r = m;
	while (l < r) {
		int mid = l + (r - l) / 2;
		if (check(mid)) {
			r = mid;
		} else {
			l = mid + 1;
		}
	}// 二分找到哪一天会出现不够用的情况 
	if (check(r)) {// 如果爆掉，直接输出 
		cout << "-1" << endl << r;
	} else {// 如果不爆，说明都够用 
		cout << "0";
	}
	return 0;
}