题目
思路
- 起初我想用树状数组来解,用树状数组维护某个区间的最小值,因为只要最小值满足租借的条件,那么该区间的其他值也一定满足。但是问题在于树状数组维护的区间结构不是任意的,比如某一天的订单将[l,r]的教室数都进行了修改,那么此时必须要更新所有包含[l,r]区间的任一值的树状树状数组,这样其实导致复杂度并没有降低。
- 跟ACMer大佬叹宝讨论后才知道,其实线段树比树状数组的能力要强大很多,比如对于这道题线段树是可以解的,但是树状数组却不行。所以树状数组更容易求解的实际上是从头开始计算并维护的一些东西,比如前缀和。
- 其实这道题的本意还是让用差分+二分来做,下面解释为什么可以用差分+二分。
差分数组
差分数组初始化所有的值为0,也就是认为a[i]=a[j],i,j任意
差分数组定义:dif[i]=a[i]-a[i-1]
差分数组可以快速进行区间修改,比如将a[l,r]区间的每个元素的值+x,那么实际上就是
dif[l]+=x;
dif[r+1]-=x;
这样区间修改的复杂度变为O(1),单点、区间查询的复杂度为O(n),因为查询时要从头开始计算
利用差分数组,此题可以在O(k+n)复杂度判断前k个订单是否有问题,因为需要k次区间修改,一次区间查询。
二分
- 二分应用的条件为:满足单调性或可以舍去局部。
- 对于这题来说是可以舍弃局部,因为如果我们发现了前mid次订单有问题,那么下次我们可以舍弃mid后面的订单,而在[1,mid]内进行搜索,否则就在[mid+1,m]内进行搜索。
- 总复杂度就变成了O((n+m)logn)
AC代码
#include<iostream>
#include<algorithm>
#include<string.h>
#include<cstring>
using namespace std;
int n,m;
int r[1000001];
struct dd{
int d;
int s;
int t;
}D[1000001];
//差分
int dif[1000001];
int need[1000001];
bool isok(int x){
bool result = true;
memset(dif,0,(n+1)*sizeof(int));
memset(need,0,(n+1)*sizeof(int));
for(int i=1;i<=x;i++){
dif[D[i].s]+=D[i].d;
dif[D[i].t+1]-=D[i].d;
}
for(int i=1;i<=n;i++){
need[i] = need[i-1]+dif[i];
if(need[i]>r[i]){
result = false;
break;
}
}
return result;
}
int main(){
cin>>n>>m;
int ans = 0;
for(int i=1;i<=n;i++) cin>>r[i];
for(int i=1;i<=m;i++) cin>>D[i].d>>D[i].s>>D[i].t;
if(isok(m)){
cout<<ans;
}else{
int l = 1;
int r = m;
int mid;
while(l<r){
mid = (l+r)/2;
if(isok(mid)){
l = mid+1;
}else{
r = mid;
}
}
cout<<-1<<'\n'<<r;
}
}