题目
在大学期间,经常需要租借教室。
大到院系举办活动,小到学习小组自习讨论,都需要向学校申请借教室。
教室的大小功能不同,借教室人的身份不同,借教室的手续也不一样。
面对海量租借教室的信息,我们自然希望编程解决这个问题。
我们需要处理接下来 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
题解
import java.util.Arrays;
import java.util.Scanner;/*** @author akuya* @create 2024-02-29-16:59*/
public class Main {static int N=1000010;static int M=N;static int n,m;static int num[];static int l[]=new int[M];static int r[]=new int[M];static long b[]=new long[N];static int v[]=new int[N];public static void main(String[] args) {Scanner scanner=new Scanner(System.in);n=scanner.nextInt();m=scanner.nextInt();num=new int[n+1];for(int i=1;i<=n;i++){num[i]=scanner.nextInt();}for(int i=1;i<=m;i++){v[i]=scanner.nextInt();l[i]=scanner.nextInt();r[i]=scanner.nextInt();}int l=0;int r=m;while(l<r){int mid=l+r+1>>1;if(check(mid)) l=mid;else r=mid-1;}if(r==m) System.out.println(0);else{System.out.println("-1");System.out.println(r+1);}}public static boolean check(int mid){Arrays.fill(b,0);for(int i=1;i<=mid;i++){b[l[i]]+=v[i];b[r[i]+1]-=v[i];}for(int i=1;i<=n;i++){b[i]+=b[i-1];if(b[i]>num[i]){return false;}}return true;}
}
思路
这道题运用了前缀和与拆分的思想,通过二分查找mid及以前的天数是否会让教室不够,如果最后r=请求数,则能完成数量要求,不等于则输出r+1,为导致超出预计的编号。二分时间复杂度logn,拆分时间复杂度为n,时间复杂度为nlogn,刚好满足时间复杂度。