洛谷P1083 [NOIP2012 提高组] 借教室

一、分析

对题面进行分析可以知道，每次修改其实都是对一个区间进行一次更改，也就是把这个区间从左到右的每一个数字加x或者减去x。如果我们学习过差分数组算法，我们应该知道，对一个区间的所有数字加一个数字或者减一个数字，它的差分数组只有最左边增加（减少）了x，最右边的后一位差分数组减少（增加）x。代码实现是：diff[l]+=x; diff[r+1]-=x; 依靠差分数组，我们可以知道每一天教室的需求量（need[i]=diff[i]+need[i-1]），如果有某一份订单的需求量大于供应量，那么肯定有订单无法满足，那么我们就得找出具体是哪一份订单不满足要求。

那么我们要怎么找出是哪一份订单不满足要求呢？

我们知道一共有m条订单，如果第mid条满足要求，那么mid肯定就要往靠近m的方向增加，反之就要向远离m的方向减小。所以我们能够想到这里应该使用二分查找算法，如果1~x条订单都满足要求，那么l=mid+1；否则r=mid-1;最后返回l的值。（l的值就是不满足的那一份订单）

AC Code:

#include<bits/stdc++.h>
using namespace std;
#define int long long
int n, m;
const int N = 1e6 + 10;
int a[N];
int d[N], s[N], t[N];
int need[N];

bool check(int x) {
    vector<int>diff(N);//差分数组

    for (int i = 1; i <= x; i++) {//x条订单-->x次修改
        diff[s[i]] += d[i];
        diff[t[i] + 1] -= d[i];
    }

    for (int i = 1; i <= n; i++) {//n天
        need[i] = diff[i] + need[i - 1];
        if (need[i] > a[i]) {
            return false;
        }
    }

    diff.clear();

    return true;
}

signed main() {
    ios::sync_with_stdio(0), cin.tie(0), cout.tie(0);

    cin >> n >> m;
    for (int i = 1; i <= n; i++) {
        cin >> a[i];
    }

    for (int i = 1; i <= m; i++) {
        cin >> d[i] >> s[i] >> t[i];
    }

    if (check(m)) {//如果所有的订单都满足
        cout << 0;
        return 0;
    }

    //如果并不是所有的订单都能满足，那么二分查找是哪一份订单（哪一次修改）不满足
    int l = 1, r = m;
    while (l <= r) {
        int mid = l + r >> 1;
        if (check(mid)) {//如果1~mid份订单都满足，那就往右移动
            l = mid + 1;
        }
        else {//如果不满足了
            r = mid - 1;
        }
    }

    cout << -1 << '\n' << l;

    return 0;
}