题目描述
在大学期间,经常需要租借教室。大到院系举办活动,小到学习小组自习讨论,都需要向学校申请借教室。教室的大小功能不同,借教室人的身份不同,借教室的手续也不一样。
面对海量租借教室的信息,我们自然希望编程解决这个问题。
我们需要处理接下来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,第二行输出需要修改订单的申请人编号。
输入样例
4 3
2 5 4 3
2 1 3
3 2 4
4 2 4
输出样例
-1
2
解题思路:二分+差分
用二分的思路,假设mid为要修改的人的前一个人,然后判断这mid是否符合条件,若符合则l=mid+1,否则r=mid-1,在判断mid是否符合条件时,由于他在sj到tj他们要借教室,从sj到tj这个区间的教室要相应减少,这时候就想到用差分数组delta维护。由于r数组装的是每天所有的教室个数,所以delta[i] = r[i] - r[i - 1],由于mid为要修改的人的前一个人,所以从1到mid的订单都满足,所以delta数组的相应区间要进行减少,操作完之后,对差分数组求和,判断得到的结果的正负,正就为真,负就为假。在输出时,如果所有订单都满足的话,l=mid+1,此时就输出0,否则就输出l。
AC代码
#include<bits/stdc++.h>
using namespace std;
int r[1000100];
int n, m;
struct ty
{
int d;
int s;
int t;
}a[1000100];
int delta[1000100];//差分数组,记录当天与前一天ri的差值
int judge(int mid)
{
memset(delta, 0, sizeof(delta));
for (int i = 1; i <= n; i++)
{
delta[i] = r[i] - r[i - 1];
}
for (int i = 1; i <= mid; i++)
{
delta[a[i].s] -= a[i].d;
delta[a[i].t + 1] += a[i].d;
}
int sum = 0;
for (int i = 1; i <=n; i++)//差分数组求和
{
sum += delta[i];
if (sum < 0)
return 0;
}
return 1;
}
int main()
{
scanf("%d%d", &n, &m);//n, m,表示天数和订单的数量。
for (int i = 1; i <= n; i++)
scanf("%d", &r[i]);
for (int i = 1; i <= m; i++)
scanf("%d%d%d", &a[i].d, &a[i].s, &a[i].t);
int l = 0, r = m;//右边界是最后一个订单
while (l <= r)
{
int mid = (l + r) >> 1;//mid是假设的需要修改订单的前一个人
if (judge(mid))
l = mid + 1;
else
r = mid - 1;
}
if (l > m)cout << 0;
else cout << "-1\n" << l;
return 0;
}