题目描述
在大学期间,经常需要租借教室。大到院系举办活动,小到学习小组自习讨论,都需要向学校申请借教室。教室的大小功能不同,借教室人的身份不同,借教室的手续也不一样。
面对海量租借教室的信息,我们自然希望编程解决这个问题。我们需要处理接下来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开始的整数编号。
对于10%的数据,有1≤ n,m≤ 10;
对于30%的数据,有1≤ n,m≤1000;
对于70%的数据,有1≤ n,m≤ 105;
对于100%的数据,有1≤n,m≤106, 0≤ri,dj≤109, 1≤sj≤tj≤n。
输出
如果所有订单均可满足,则输出只有一行,包含一个整数0。否则(订单无法完全满足)输出两行,第一行输出一个负整数-1,第二行输出需要修改订单的申请人编号。
样例输入
4 3
2 5 4 3
2 1 3
3 2 4
4 2 4
样例输出
-12
解题思路:这道题目如果数据较小,可以直接暴力解(如果那样,就不会出第二题了)。由于数据量较大,如果暴力的话只能得30分。网上有线段树的方法,这里使用二分的方法来写,比较容易理解。设mid=(0+(m+1))。先考虑1到mid份订单的合法性,如果1到mid合法的话,再考虑1到(mid+(m+1)/2)份订单,如果不合法,则考虑1到(1+mid/2)份订单,依次类推,最终即可求得结果。注意在judge中求和采用了一些技巧来提升效率(原理见最下方图)。
参考程序:
#include<iostream>
#include<algorithm>
#include<cstring>
#include<cstdio>
using namespace std;
const int N=1000005;
int room[N],d[N],s[N],t[N],sum[N];
long long a[N];
int n,m;
void init()
{
scanf("%d%d",&n,&m);//cin>>n>>m;
for (int i=1;i<=n;i++) scanf("%d",&room[i]);//cin>>room[i];
for (int i=1;i<=m;i++) scanf("%d%d%d",&d[i],&s[i],&t[i]);//cin>>d[i]>>s[i]>>t[i];
}
bool judge(int ad){
memset(a,0,sizeof(a));
for (int i=1;i<=ad;i++){
a[s[i]]+=d[i];
a[t[i]+1]-=d[i];
}
long long num(0);
for (int i=1;i<=n;i++){
num+=a[i];
if (num>room[i]) return false;
}
return true;
}
void work(){
int l=0,r=m+1;
while (l+1<r){
int mid=(l+r)/2;
if (judge(mid))
l=mid;
else
r=mid;
}
if (l==m) printf("0\n");
else printf("-1\n%d\n",l+1);
}
int main()
{
init();
work();
return 0;
}
2 | 5 | 4 | 3 | |||
0 | 1 | 2 | 3 | 4 | 5 | |
a[] | 0 | 2 | (3+4) | 0 | -2 | (-3-4) |
num | 0 | 2 | 9 | 9 | 7 | 0 |