此题是典型的区间修改问题,采用差分数组进行优化。
在判断是否满足条件时,采用的是新构建一个need数组的方法,比较其与要求教室的数量大小,而不是作差判负。
采用二分法进行优化。老样子,注意边界条件。mid = (l + r) / 2,而不是 mid = (l + r + 1) / 2,具体如何选择,取决于边界收缩条件,本题中l = mid +1 , r = mid 决定了不能是 mid = (l + r + 1) / 2,否则有可能会陷入死循环(对所有测试数据而言,有的测试数据不会死循环,有的会)。
选择 mid = (l + r + 1) / 2 ,请参考我的这篇博客。
#include<iostream>
#include<cstring>
#include<cstdio>
using namespace std;
int n,m;
int diff[1000011],need[1000011],rest[1000011],r[1000011],l[1000011],d[1000011];
bool isok(int x)
{
memset(diff,0,sizeof(diff));
for(int i=1;i<=x;i++)
{
diff[l[i]]+=d[i];
diff[r[i]+1]-=d[i];
}
for(int i=1;i<=n;i++)
{
need[i]=need[i-1]+diff[i];
if(need[i]>rest[i])return 0;
}
return 1;
}
int main()
{
scanf("%d%d",&n,&m);
for(int i=1;i<=n;i++)scanf("%d",&rest[i]);
for(int i=1;i<=m;i++)scanf("%d%d%d",&d[i],&l[i],&r[i]);
int begin=1,end=m;
if(isok(m)){cout<<"0";return 0;}
while(begin<end)
{
int mid=(begin+end)/2;
if(isok(mid))begin=mid+1;
else end=mid;
}
cout<<"-1"<<endl<<begin;
}