Acwing 503. 借教室

在这里插入图片描述
在这里插入图片描述
这道题我刚开始也是想用差分直接解决,但是好像时间复杂度是o(n^2)级别的
所以用了一个二分+差分的做法
代码如下

#include<iostream>
using namespace std;
const int N=1e6+10;
int a[N],b[N],c[N];
int s[N],z[N],y[N];
int n,m;
bool check(int mid)
{
    
    
    for(int i=1;i<=n;i++) y[i]=s[i];
    for(int i=1;i<=mid;i++)
    {
    
    
        y[a[i]]-=c[i];
        y[b[i]+1]+=c[i];
    }
    long long sum=0;
    for(int i=1;i<=n;i++)
    {
    
    
        sum+=y[i];
        if(sum<0) return true;
    }
    return false;
}
int main(void)
{
    
    
    cin>>n>>m;
    for(int i=1;i<=n;i++) scanf("%d",&z[i]),s[i]=z[i]-z[i-1];
    
    for(int i=1;i<=m;i++) scanf("%d%d%d",&c[i],&a[i],&b[i]);
    
    int l=0,r=m;
    while(l<r)
    {
    
    
        int mid=l+r>>1;
        if(check(mid)) r=mid;
        else
        l=mid+1;
    }
    if(check(l))
    cout<<"-1"<<endl<<l;
    else
    cout<<"0";
}

猜你喜欢

转载自blog.csdn.net/qq_52358098/article/details/113714618