在大学期间,经常需要租借教室。
大到院系举办活动,小到学习小组自习讨论,都需要向学校申请借教室。
教室的大小功能不同,借教室人的身份不同,借教室的手续也不一样。
面对海量租借教室的信息,我们自然希望编程解决这个问题。
我们需要处理接下来 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 个教室,而它们具体是哪些教室,每天是否是相同的教室则不用考虑。
借教室的原则是先到先得,也就是说我们要按照订单的先后顺序依次为每份订单分配教室。
如果在分配的过程中遇到一份订单无法完全满足,则需要停止教室的分配,通知当前申请人修改订单。
这里的无法满足指从第 s j s_j sj 天到第 t j t_j tj 天中有至少一天剩余的教室数量不足 d j d_j dj 个。
现在我们需要知道,是否会有订单无法完全满足。
如果有,需要通知哪一个申请人修改订单。
输入格式
第一行包含两个正整数
n
,
m
n,m
n,m,表示天数和订单的数量。
第二行包含
n
n
n 个正整数,其中第
i
i
i 个数为
r
i
r_i
ri,表示第
i
i
i 天可用于租借的教室数量。
接下来有
m
m
m 行,每行包含三个正整数
d
j
,
s
j
,
t
j
d_j,s_j,t_j
dj,sj,tj,表示租借的数量,租借开始、结束分别在第几天。
每行相邻的两个数之间均用一个空格隔开。
天数与订单均用从
1
1
1 开始的整数编号。
输出格式
如果所有订单均可满足,则输出只有一行,包含一个整数 0
。
否则(订单无法完全满足)输出两行,第一行输出一个负整数 −1
,第二行输出需要修改订单的申请人编号。
数据范围
1
≤
n
,
m
≤
1
0
6
,
1≤n,m≤10^6,
1≤n,m≤106,
0
≤
r
i
,
d
j
≤
1
0
9
,
0≤ri,dj≤10^9,
0≤ri,dj≤109,
1
≤
s
j
≤
t
j
≤
n
1≤sj≤tj≤n
1≤sj≤tj≤n
输入样例:
4 3
2 5 4 3
2 1 3
3 2 4
4 2 4
输出样例:
-1
2
首先根据题目,我们需要实现对于一段区域的加上一个数的操作,如果直接暴力求解必然会超时,故可以使用差分数组的做法。
其次,题目中要求,如果订单有误,那么就要求出第一个错误的订单,那么可以使用二分来搜最后一个正确的订单或者搜第一个错误的订单。
对于差分的详细复习与对二分方法的详细解释:2024蓝桥杯每日一题-借教室
#include<iostream>
#include<cstring>
using namespace std;
const int N = 1e6 + 10;
#define ll long long
int n, m;
int w[N];
int d[N], s[N], t[N];
ll b[N];
bool check(int mid) {
memset(b, 0, sizeof b); //先初始化差分数组
for (int i = 1; i <= mid; i++) {//差分模板,处理前mid个订单
b[s[i]] += d[i];
b[t[i] + 1] -= d[i];
}
ll s = 0;
for (int i = 1; i <= n; i++) { //枚举每一天
s += b[i]; //差分对应的前缀和,即加完d[i]的w[]数组
if (s > w[i])return false; //如果有一天无法满足订单,就返回 false
}
return true;
}
int main() {
cin >> n >> m;
for (int i = 1; i <= n; i++) cin >> w[i];
for (int i = 1; i <= m; i++)cin >> d[i] >> s[i] >> t[i];
//这里使用二分去查找能够满足的最后一个订单
int l = 0, r = m;
while (l < r) {
int mid = l + r + 1 >> 1;
if (check(mid)) l = mid;
else r = mid - 1;
}
if (r == m)cout << 0 << endl; //如果全部都可以满足的话输出 0
else cout << -1 << endl << r + 1 << endl; //输出 r+1 是因为二分找到的r是能够满足的最后一个订单
return 0;
}
另一种写法:找第一个不能满足的订单
#include<iostream>
#include<cstring>
using namespace std;
const int N = 1e6 + 10;
#define ll long long
int n, m;
int w[N];
int d[N], s[N], t[N];
ll b[N];
bool check(int mid) {
memset(b, 0, sizeof b); //先初始化差分数组
for (int i = 1; i <= mid; i++) {//差分模板,处理前mid个订单
b[s[i]] += d[i];
b[t[i] + 1] -= d[i];
}
ll s = 0;
for (int i = 1; i <= n; i++) { //枚举每一天
s += b[i]; //差分对应的前缀和,即加完d[i]的w[]数组
if (s > w[i])return true; //如果有一天无法满足订单,就返回true,因为找的就是不能满足的
}
return false;
}
int main() {
cin >> n >> m;
for (int i = 1; i <= n; i++) cin >> w[i];
for (int i = 1; i <= m; i++)cin >> d[i] >> s[i] >> t[i];
//这里使用二分去查找不能满足的第一个订单
int l = 0, r = m;
while (l < r) {
int mid = l + r >> 1;
if (check(mid)) r = mid;
else l = mid + 1;
}
if (!check(r))cout << 0 << endl; //如果全部都可以满足的话输出0
else cout << -1 << endl << r << endl; //输出r,即第一个不满足的订单
return 0;
}