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
二分查找+差分
#include<bits/stdc++.h>
using namespace std;
#define ll long long
const int z=1e6+10; // 设定一个大数组的范围,用于存储数据
ll a[z],b[z]; // a数组用来存放每天可租借教室的数量,b是差分数组
ll c[z],c_l[z],c_r[z]; // c数组存放订单的教室数量,c_l和c_r分别存放订单的开始和结束天
ll copy_a[z]; // copy_a数组用于在check函数中复制a数组,以便进行计算
int n,m,i; // n表示天数,m表示订单数量,i用于循环
// check函数检查是否能满足前mid个订单
bool check(ll mid)
{
for(i=1;i<=n;i++)
{
b[i]=a[i]-a[i-1]; // 初始化差分数组b
copy_a[i]=a[i]; // 复制a数组到copy_a,用于计算是否能满足订单
}
for(i=1;i<=mid;i++)
{
b[c_l[i]]-=c[i]; // 对于每个订单,减去开始租借的教室数量
b[c_r[i]+1]+=c[i]; // 在结束租借的下一天,加回教室数量
}
for(i=1;i<=n;i++)
{
copy_a[i]=copy_a[i-1]+b[i]; // 从差分数组还原出实际每天可用的教室数量
if(copy_a[i]<0) return true; // 如果某天教室数量不足,返回true
}
return false; // 所有订单都能满足,返回false
}
int main()
{
scanf("%d %d",&n,&m); // 读取天数n和订单数m
for(i=1;i<=n;i++)
scanf("%lld",&a[i]); // 读取每天可租借的教室数量
for(i=1;i<=m;i++)
scanf("%lld %lld %lld",&c[i],&c_l[i],&c_r[i]); // 读取每个订单的详情
// 检查是否所有订单均能满足
if(!check(m))
{
printf("0"); // 如果能满足,输出0
return 0;
}
// 二分查找无法满足的订单位置,左查找,查找左边界
ll l=1,r=m; // 设置检查的范围
ll mid; // 用于存储中间值
while(l<r)
{
mid=(l+r)>>1; // 找到中间值
if(check(mid)) r=mid; // 如果前mid个订单无法满足,减小右边界
else l=mid+1; // 如果能满足,增大左边界
}
printf("-1\n%d",r); // 输出-1和无法满足的订单编号
return 0;
}
在这段代码中,我们首先使用差分数组来处理订单的教室租借问题。差分数组是一种处理数据变化的技巧,特别适合处理区间增减操作。通过对差分数组进行前缀和操作,我们可以还原出每天实际可用的教室数量,并检查是否所有的订单都可以被满足。
如果不能满足所有订单,我们使用二分查找来确定第一个无法满足的订单。这是通过不断缩小检查的订单范围实现的,直到找到一个点,在这一点上,如果包括这个订单,就无法满足所有订单的需求。这样做的目的是找到那个特定的订单,我们需要告知申请者修改他们的订单。