【前缀和与差分 二分搜索 C/C++】洛谷 P1083 借教室

- 第 79 篇 -
Date: 2025 - 03 - 11
Author: 郑龙浩/仟濹
【前缀和与差分 二分搜索 C/C++】

已总结一维前缀和一维差分 相关内容,详见我的博客:
一维前缀和一维差分

已总结一维前缀和一维差分 相关内容,详见我的博客:

文章目录

洛谷 P1083 借教室

题目描述

在大学期间,经常需要租借教室。大到院系举办活动,小到学习小组自习讨论,都需要向学校申请借教室。教室的大小功能不同,借教室人的身份不同,借教室的手续也不一样。

面对海量租借教室的信息,我们自然希望编程解决这个问题。

我们需要处理接下来 n n n 天的借教室信息,其中第 i i i 天学校有 r i r_i ri 个教室可供租借。共有 m m m 份订单,每份订单用三个正整数描述,分别为 d j , s j , t j d_j,s_j,t_j dj,sj,tj,表示某租借者需要从第 s j s_j sj 天到第 t j t_j tj 天租借教室(包括第 s j s_j sj 天和第 t j t_j tj 天),每天需要租借 d j d_j dj 个教室。

我们假定,租借者对教室的大小、地点没有要求。即对于每份订单,我们只需要每天提供 d j d_j dj 个教室,而它们具体是哪些教室,每天是否是相同的教室则不用考虑。

借教室的原则是先到先得,也就是说我们要按照订单的先后顺序依次为每份订单分配教室。如果在分配的过程中遇到一份订单无法完全满足,则需要停止教室的分配,通知当前申请人修改订单。这里的无法满足指从第 s j s_j sj 天到第 t j t_j tj 天中有至少一天剩余的教室数量不足 d j d_j dj 个。

现在我们需要知道,是否会有订单无法完全满足。如果有,需要通知哪一个申请人修改订单。

输入格式

第一行包含两个正整数 n , m n,m n,m,表示天数和订单的数量。

第二行包含 n n n 个正整数,其中第 i i i 个数为 r i r_i ri,表示第 i i i 天可用于租借的教室数量。

接下来有 m m m 行,每行包含三个正整数 d j , s j , t j d_j,s_j,t_j dj,sj,tj,表示租借的数量,租借开始、结束分别在第几天。

每行相邻的两个数之间均用一个空格隔开。天数与订单均用从 1 1 1 开始的整数编号。

输出格式

如果所有订单均可满足,则输出只有一行,包含一个整数 0 0 0。否则(订单无法完全满足)

输出两行,第一行输出一个负整数 − 1 -1 −1,第二行输出需要修改订单的申请人编号。

输入输出样例 #1

输入 #1

复制代码
4 3 
2 5 4 3 
2 1 3 
3 2 4 
4 2 4

输出 #1

复制代码
-1 
2

说明/提示

【输入输出样例说明】

1 份订单满足后,4 天剩余的教室数分别为 0 , 3 , 2 , 3 0,3,2,3 0,3,2,3。第 2 2 2 份订单要求第 2 天到第 4 4 4 天每天提供 3 3 3 个教室,而第 3 3 3 天剩余的教室数为 2,因此无法满足。分配停止,通知第 2 2 2 个申请人修改订单。

【数据范围】

对于 10 % 10\% 10% 的数据,有 1 ≤ n , m ≤ 10 1\le n,m\le 10 1≤n,m≤10;

对于 30 % 30\% 30% 的数据,有 1 ≤ n , m ≤ 1000 1\le n,m\le 1000 1≤n,m≤1000;

对于 70 % 70\% 70% 的数据,有 1 ≤ n , m ≤ 1 0 5 1 \le n,m \le 10^5 1≤n,m≤105;

对于 100 % 100\% 100% 的数据,有 1 ≤ n , m ≤ 1 0 6 1 \le n,m \le 10^6 1≤n,m≤106, 0 ≤ r i , d j ≤ 1 0 9 0 \le r_i,d_j\le 10^9 0≤ri,dj≤109, 1 ≤ s j ≤ t j ≤ n 1 \le s_j\le t_j\le n 1≤sj≤tj≤n。

NOIP 2012 提高组 第二天 第二题

2022.2.20 新增一组 hack 数据

思路

算法: 一维差分, 一维前缀和, 二分查找

利用 一维差分 与 一维前缀和 对 **第 i 天 ++需要++ 租借教室的数量(差分数组)**进行 差分标记 ,可以得出对应 区间标记,然后进行前缀和计算,得出第 i 天教室需要租借多少教室。

在前缀和的过程中进行判断,

  • 如果 在前缀和的过程中 第 i 天 需要租借数量 > 可租借数量,则表示租借不了,此时直接返回错误提示,则不进行后续的前缀和操作。

    然后,利用 二分搜索 找到 是哪一个订单超出了可租借的数量

  • 如果在前缀和的过程中,所有第 i 天的租借数量 < 可租借数量,则表示每天都可以租借,直接打印答案为 0

代码

cpp 复制代码
// 洛谷P1083借教室
// Author: 郑龙浩 / 仟濹
// Date: 2025-03-11
#include <bits/stdc++.h>
using namespace std;
int days, num; // 天数 订单数量

vector <long long> diff, arr;// 每天需要租借教室的数量, 每天可借最高数量
vector <int> l, r, value; // 借教室的起点, 借教室的终点,每天可以租借教室的数量
// 检查 第 1 天 ~ 第 n 天 是否满足所有订单,满足返回 1,不满足返回0
// 实际上就是对差分数组进行前缀和计算,如果有 某一天不满足订单数量,则返回0
// 如果每天都满足订单数量,则返回 1
bool check(int n){
    fill(diff.begin(), diff.end(), 0);

    // 差分标记 -- 订单数量多少,就标记多少
    // 前 n 个订单的差分操作
    for (int i = 1; i <= n; i++){
        diff[ l[i] ] += value[i];
        if (r[i] + 1 <= days) diff[r[i] + 1] -= value[i]; 
    }
    
    // 对进行差分标记后的数组进行前缀和计算 --> 得出每天应该借多少个教室
    for (int i = 1; i <= days; i++){
        diff[i] += diff[i - 1]; // 计算出当天的订单量
        // 如果不满足订单量,即所需租借教室的数量,则返回0
        if (diff[i] > arr[i])
            return false;
    }
    // 如果每天的可租借教室数量满足订单,那么返回1
    return true;
}
int main( void ){
    cin >> days >> num;
    diff.resize(days + 2); // 差分数组开辟的空间要大一个
    arr.resize(days + 1); // 存储每天可以租借的教室 -- 即每天可以租借教室的上限
    
    l.resize(num + 1);
    r.resize(num + 1);
    value.resize(num + 1);
    for (int i = 1; i <= days; i++) // 输入每天可以租借教室的上限
        cin >> arr[i];
    // 输入 租借数量 租借开始 结束租借
    for (int i = 1; i <= num; i++)
        cin >> value[i] >> l[i] >> r[i];

    // 检查 1 ~ num 的订单是否出错
    if (check(num)){
        cout << 0;
        return 0;
    } 

    int begin = 1, end = num; // 查找边界 --- 左边界 --- 右边界
    int ans; // 存储答案
    // 如果不能满足订单数量,则寻找是哪个(哪天的)订单不能满足
    while (begin <= end){
        int mid = (begin + end) / 2;
        // 如果满足订单,则可以尝试寻找更靠后的订单
        if ( check(mid) ) 
            begin = mid + 1;
        // 如果不满足订单,则必须寻找更靠前的订单
        else {
            ans = mid;
            end = mid - 1;
        }
    }
    //为什么是begin呢???
    cout << -1 << endl << ans << endl;
    return 0;
}
相关推荐
Swift社区24 分钟前
LeetCode 高频题实战:如何优雅地序列化和反序列化字符串数组?
算法·leetcode·职场和发展
醍醐三叶39 分钟前
C++核心编程--3 函数提高
c++
徐子童1 小时前
《从零开始入门递归算法:搜索与回溯的核心思想 + 剑指Offer+leetcode高频面试题实战(含可视化图解)》
算法
天宫风子1 小时前
抽象代数小述(二之前)
经验分享·笔记·算法·生活·抽象代数
向上的车轮1 小时前
“傅里叶变换算法”来检测纸箱变形的简单示例
算法
范纹杉想快点毕业2 小时前
以项目的方式学QT开发(一)——超详细讲解(120000多字详细讲解,涵盖qt大量知识)逐步更新!
c语言·数据结构·c++·git·qt·链表·github
九亿AI算法优化工作室&2 小时前
乡村地区无人机医药配送路径规划与优化仿真
人工智能·算法·matlab·回归
米粉03052 小时前
算法图表总结:查找、排序与递归(含 Mermaid 图示)
数据结构·算法·排序算法
人类发明了工具2 小时前
【优化算法】协方差矩阵自适应进化策略(Covariance Matrix Adaptation Evolution Strategy,CMA-ES)
线性代数·算法·矩阵·cma-es
黑色的山岗在沉睡2 小时前
LeetCode100.4 移动零
数据结构·算法·leetcode