当前位置:首页 > 提高组 > 2012年NOIP提高组「借教室」题目(P1083)解题思路与二分查找优化代码解析

2012年NOIP提高组「借教室」题目(P1083)解题思路与二分查找优化代码解析

4小时前

2012年NOIP提高组「借教室」题目(P1083)解题思路与二分查找优化代码解析 NOIP借教室 洛谷P1083 二分查找 差分数组 区间分配优化 第1张

一、题目解读

本题为2012年NOIP提高组中的「借教室」问题(洛谷P1083),要求处理教室借用订单的分配问题。给定n天每天可用教室数量r和m个订单(订单包含需求教室数d、开始日期s、结束日期t),判断能否通过删除部分订单使所有订单满足教室容量限制。若能,输出需要删除的最少订单数;若所有订单必须保留,输出-1。题目核心在于将动态分配转化为判定性问题,通过二分查找优化求解。

二、解题思路

用户提供的代码采用二分查找解决。思路是将订单数作为二分范围,通过check函数判定是否存在可行解:

1. 转化为判定性问题:若删除前mid个订单能满足,则答案在[1, mid-1]中;否则在[mid+1, m]中。

2. 差分数组优化:用diff数组记录订单对每天教室数的增量(开始日+需求,结束日-需求),避免逐日模拟

3. 边界处理与溢出防护:使用long long防止数据溢出,特判所有订单可满足的情况,避免二分循环。

三、解题步骤

1. 输入处理:读取n、m,教室容量r和订单信息d、s、t。

2. 二分查找框架:初始化left=1, right=m,答案ans初始化为m(最坏情况全删)。

3. 判定函数check(mid):

    清空diff数组,遍历前mid个订单,按区间更新差分。

    计算每天累计需求current,若超过当日容量r,返回false。

4. 二分循环:根据check结果调整左右边界,最终ans为不可行的临界点。

5. 输出结果:特判全满足输出0,否则输出ans。

四、代码与注释

#include <iostream>
#include <vector>
#include <cstring>
using namespace std;

const int MAXN = 1e6 + 10;

int n, m;
int r[MAXN]; // 每天可用教室数
int d[MAXN], s[MAXN], t[MAXN]; // 订单信息
long long diff[MAXN]; // 使用long long防止溢出

bool check(int mid) {
    memset(diff, 0, sizeof(diff)); // 更安全初始化
    for (int i = 1; i <= mid; ++i) {
        diff[s[i]] += d[i];
        if (t[i] + 1 <= n) diff[t[i] + 1] -= d[i]; // 区间右端点差分
    }
    long long current = 0;
    for (int i = 1; i <= n; ++i) {
        current += diff[i];
        if (current > r[i]) return false; // 超容量即不可行
    }
    return true;
}

int main() {
    ios::sync_with_stdio(false);
    cin.tie(0);
    cin >> n >> m;
    for (int i = 1; i <= n; ++i) cin >> r[i];
    for (int i = 1; i <= m; ++i) cin >> d[i] >> s[i] >> t[i];
    if (check(m)) { // 特判全满足
        cout << 0 << endl;
        return 0;
    }
    int left = 1, right = m, ans = m;
    while (left <= right) {
        int mid = (left + right) >> 1;
        if (!check(mid)) {
            ans = mid;
            right = mid - 1;
        } else {
            left = mid + 1;
        }
    }
    cout << -1 << endl << ans << endl;
    return 0;
}

五、总结

该代码通过二分查找将订单分配问题转化为判定性问题,结合差分数组高效处理区间影响,避免了O(nm)的暴力复杂度。关键点在于:

1. 利用二分查找缩小答案范围,降低时间复杂度至O(mlogm)。

2. 差分数组减少冗余计算,确保判定过程线性时间。

3. 特判与溢出防护提升代码鲁棒性。

此解法适用于区间修改与查询的判定场景,是竞赛中优化动态规划的经典技巧。


原创内容 转载请注明出处

分享给朋友:

发表评论

访客

看不清,换一张

◎欢迎参与讨论,请在这里发表您的看法和观点。