差分 + 二分 (借教室)代码

方糖--C++菜鸟程序员

已于 2024-01-23 23:09:19 修改

阅读量366

点赞数 10

分类专栏：刷题代码记录文章标签：算法 c++

于 2024-01-23 23:08:56 首次发布

本文链接：https://blog.youkuaiyun.com/m0_72193120/article/details/135796470

版权

刷题代码记录专栏收录该内容

20 篇文章

订阅专栏

本文介绍了一道洛谷编程题，涉及如何利用前缀和、差分数组和二分搜索技术，判断在给定教室容量限制下，能否成功处理一定数量的订单。主要展示了如何通过编程实现查询功能并找到最小满足条件的订单数量。

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >

洛谷题目：借教室

好像前缀和、差分很喜欢和二分结合

#include<iostream>
#include<cstring>
using namespace std;
//链接：
const int N = 1e6 + 10;
int rooms[N];
int subRooms[N];
int d[N],s[N],t[N];
int n,m;

bool check(int ordNum)//查询前ordNum个订单能否成功
{
    for (int i = 1; i <= n; i++)
    {
        subRooms[i] = rooms[i] - rooms[i-1];
    }
    for (int i = 1; i <= ordNum; i++)
    {
        subRooms[s[i]] -= d[i];
        subRooms[t[i]+1] += d[i];
    }
    int r = 0;
    for (int i = 1; i <= n; i++)
    {
        r += subRooms[i];
        if(r<0)
            return false;
    }
    return true;
}

int main()
{
    cin >>n >>m;
    for (int i = 1; i <= n; i++)
    {
        cin >> rooms[i];
    }
    for (int i = 1; i <= m; i++)
    {
        cin >> d[i] >> s[i] >> t[i];
    }
    
    if(check(m))
    {
        cout<<0<<endl;
        return 0;
    }
    else
        cout<<-1<<endl;
    
    int min = 1;
    int max = m;
    int mid;
    while(min <= max)
    {
        mid = min + (max-min)/2;
        if(check(mid))
        {
            min=mid+1;
        }
        else
        {
            if(max==mid)
                break;//说明找到了
            max=mid;
        }
    }

    cout << max;
}