链接:https://ac.nowcoder.com/acm/problem/16564
来源:牛客网
题目描述
在大学期间,经常需要租借教室。大到院系举办活动,小到学习小组自习讨论,都需要向学校申请借教室。教室的大小功能不同,借教室人的身份不同,借教室的手续也不一样。
面对海量租借教室的信息,我们自然希望编程解决这个问题。
我们需要处理接下来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
输入
4 3
2 5 4 3
2 1 3
3 2 4
4 2 4
输出
-1
2
说明
第1 份订单满足后,4 天剩余的教室数分别为0,3,2,3。
第2 份订单要求第2 天到第4 天每天提供3 个教室,而第3 天剩余的教室数为2,因此无法满足。分配停止,通知第2个申请人修改订单。
——————
首先二分天数,然后在进行差分,最后前缀和检查是否满足条件
#include<iostream>
#include<cstring>
#include<algorithm>
using namespace std;
typedef long long LL;
const int N=1e7+7;
LL n,m,aa[N],c[N],cc[N];
struct node{
int d,s,e;//需要天数,开始时间,结束时间
}room[N];
int check(LL mid)
{
memset(c,0,sizeof c);
for(int i=1;i<=mid;i++)
{//对第 1-mid 天的需求进行差分
c[room[i].s]+=room[i].d;
c[room[i].e+1]-=room[i].d;
}
for(int i=1;i<=n;i++) cc[i]=cc[i-1]+c[i];//求每个教室应该有多少个才能满足需求
for(int i=1;i<=n;i++)
{
if(cc[i]>aa[i]) return 0;//不满足
}
return 1;
}
int main()
{
ios_base::sync_with_stdio(0);cin.tie(0);cout.tie(0);
cin>>n>>m;
for(int i=1;i<=n;i++) cin>>aa[i];//输入每天教室个数
for(int i=1;i<=m;i++) cin>>room[i].d>>room[i].s>>room[i].e;//输入每天需求
LL l=1,r=1e6;
while(l<r)//二分天数
{
LL mid=(l+r)/2;
if(check(mid)) l=mid+1;
else r=mid;
}//最后结束条件是 l==r,所以要判断一下 l是不是满足条件
if(check(l)) cout<<"0"<<endl;
else cout<<"-1"<<endl<<l<<endl;
return 0;
}