题目:
在大学期间,经常需要租借教室。
大到院系举办活动,小到学习小组自习讨论,都需要向学校申请借教室。
教室的大小功能不同,借教室人的身份不同,借教室的手续也不一样。
面对海量租借教室的信息,我们自然希望编程解决这个问题。
我们需要处理接下来 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,
第二行输出需要修改订单的申请人编号。
数据范围:
解题代码:
import java.io.BufferedReader;
import java.io.IOException;
import java.io.InputStreamReader;
import java.util.*;
public class Main {
static int n;
static int m;
static long[] days;
static long[] usedays;
static long[] startdays;
static long[] enddays;
static long[] sub;
public static void main(String[] args) throws IOException {
// 这里使用快读的方式,因为可以一次性读取多个字符,所以,比Scanner的性能较好
BufferedReader reader = new BufferedReader(new InputStreamReader(System.in));
String nm = reader.readLine();
String[] nmstr = nm.split(" ");
n = Integer.parseInt(nmstr[0]); // 天数
m = Integer.parseInt(nmstr[1]); // 总订单数
String[] day = reader.readLine().split(" ");
days = new long[n + 10]; //加10 数组下标从1开始使用
for (int i = 1; i <= n; i++) {
days[i] = Integer.parseInt(day[i - 1]);
}
// 定义三个一维数组 分别用来存放每个订单的
// 租教室的数量 开始的日期 截止的日期
// 并且同一个订单 放在三个数组的同一个下标处
usedays = new long[n + 10];
startdays = new long[n + 10];
enddays = new long[n + 10];
// 将sub 差分数组创建出来
sub = new long[n + 10];
for (int i = 1; i <= m; i++) {
String[] order = reader.readLine().split(" ");
usedays[i] = Long.parseLong(order[0]);
startdays[i] = Long.parseLong(order[1]);
enddays[i] = Long.parseLong(order[2]);
}
// 到此处为止,从键盘获取数据就完成了
int l = 0;
// 下标从1开始,但这里的l 之所以从0开始,是为了防止如果第一个订单就无法满足,
// 会出现 l == r == 1 但是下面输出是 r+1 所以会 出现结果错误
// 定义 l 是最上变的订单 r 是最下面的订单
int r = m;
// 这里使用二分法逐一查找
while (l < r) {
int mid = (l + r + 1) / 2;
if (check(mid)) {
l = mid;
} else {
r = mid - 1;
}
}
// 最后,循环结束 如果 l指的订单不是最下面的订单
// 那么说明,空教室不够所有人使用 此时 r 所指的订单就是不满足的订单
if (r == m) {
System.out.println("0");
} else {
System.out.println("-1");
System.out.println(r + 1);
}
}
private static boolean check(int mid) {
// 在这里编写判断逻辑 根据mid所指的中间值
Arrays.fill(sub, 0); // 用 0 填充差分数组
for (int i = 1; i <= mid; i++) {
sub[Math.toIntExact(startdays[i])] += usedays[i];
sub[Math.toIntExact(enddays[i] + 1)] -= usedays[i];
}
// 最上面的一个订单 其差分后的值 变大了一个usedays
// 最下面的一个订单 其差分后的值 变小了一个usedays
// 恢复成原始数组
for (int i = 1; i <= n; i++) {
sub[i] += sub[i - 1];
if (sub[i] > days[i]) {
return false;
}
}
return true;
}
}