链接:登录—专业IT笔试面试备考平台_牛客网
来源:牛客网
在大学期间,经常需要租借教室。大到院系举办活动,小到学习小组自习讨论,都需要向学校申请借教室。教室的大小功能不同,借教室人的身份不同,借教室的手续也不一样。
面对海量租借教室的信息,我们自然希望编程解决这个问题。
我们需要处理接下来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
4 3 2 5 4 3 2 1 3 3 2 4 4 2 4
输出
-1 2
-1 2
说明
第1 份订单满足后,4 天剩余的教室数分别为0,3,2,3。 第2 份订单要求第2 天到第4 天每天提供3 个教室,而第3 天剩余的教室数为2,因此无法满足。分配停止,通知第2个申请人修改订单。
首先很容易想到一个暴力算法就是我们能够通过将所有对应的天数里进行比大小,当某一天不满足时候就可以直接退出,这样下来,我们就能够得到一定的正确解。
但是,很明显无法通过。
而我们再仔细阅读题目,我们发现,我们需要将某一段区间内的一定数字进行操作,可以想到来一个前缀和差分。
我们主要是要对题目中要求的天数区间内进行数字加法操作,通过判断这些天数与我们某些天的最多数字进行比较,如果不可以我们直接跳出。
通过这样的分析,一定区间内实施加减操作,很明显可以想到差分算法。
枚举每一个d[i]与l,r对相应区间进行操作。然后我们就可以得到一个差分数组,再进行前缀和运算就可以得到一个need数组,这个数组能够存放每一天需求的数量,然后我们就可以通过比较need数组与每天的rest大小来判断。
其次,我们能想到什么?
当然是二分答案,我们可以判断每一次答案是否符合,并且前一个不符合,后面的立刻不进行操作,于是我们就可以通过二分答案来判断哪一个是最终解。
#include<bits/stdc++.h>
using namespace std;
const int N=1000010;
typedef long long ll;
ll rest[N];
ll need[N];
ll chafen[N];
ll r[N],l[N],d[N];
int n,m;
bool check(int x)
{
memset(chafen,0,sizeof(chafen));
for(int i=1;i<=x;i++)
{
chafen[l[i]]+=d[i];
chafen[r[i]+1]-=d[i];
}
for(int i=1;i<=n;i++)
{
need[i]=need[i-1]+chafen[i];
if(need[i]>rest[i])return 0;
}
return 1;
}
int main()
{
cin>>n>>m;
for(int i=1;i<=n;i++)
cin>>rest[i];
for(int i=1;i<=m;i++)scanf("%d%d%d",&d[i],&l[i],&r[i]);
if(check(m))
{
cout<<"0"<<"\n";
return 0;
}
int l=1,r=m;
while(l<r)
{
int mid=(l+r)>>1;
if(check(mid))
{
l=mid+1;
}
else
{
r=mid;
}
}
cout<<"-1"<<"\n"<<r;
return 0;
}