[NOIP2012]借教室

传送门

题目思路

刚开始是打算用线段树来维护的 后来感觉会超时
应该用二分+差分
二分的是第几次无法借出教室 通过差分来维护区间信息
与每天的教室数量比较 如果多于某天天教室数量 则r=mid
题目存在单调性 m越大 越容易不满足要求

代码

#include <bits/stdc++.h>

using namespace std;

const int N = 1e6 + 10;

int n, m;
struct Node
{
    int d, s, t;
}order[N];
int a[N], b[N];

bool check(int mid)
{
    memset(b, 0, sizeof b);
    for (int i = 1; i <= mid; i ++ )
    {
        int d = order[i].d;
        b[order[i].s] += d, b[order[i].t + 1] -= d;
    }
    for (int i = 1; i <= n; i ++ )
    {
        b[i] += b[i - 1];
        if (b[i] > a[i]) return false;
    }
    return true;
}
    
int main()
{
    scanf("%d%d", &n, &m);
    
    for (int i = 1; i <= n; i ++ ) scanf("%d", &a[i]);
    for (int i = 1; i <= m; i ++ )
    {
        int d, s, t;
        scanf("%d%d%d", &d, &s, &t);
        order[i] = {d, s, t};
    }
    
    if (check(m))
    {
        puts("0");
        return 0;
    }
    
    int l = 1, r = m;
    while (l < r)
    {
        int mid = l + r >> 1;
        if (!check(mid)) r = mid;
        else l = mid + 1;
    }
  
    printf("-1\n%d", r);
    
    return 0;
}