[NOIP2012 提高组] 借教室

最新推荐文章于 2024-05-01 18:03:45 发布

Achilles0705.

最新推荐文章于 2024-05-01 18:03:45 发布

阅读量864

点赞数 16

文章标签：算法数据结构 c++

本文链接：https://blog.csdn.net/Achilles0705/article/details/136356949

版权

文章介绍了一种使用前缀和和二分查找优化的算法，解决大学教室租借问题，以高效满足或识别无法满足的订单。

摘要由CSDN通过智能技术生成

题目描述

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

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

我们需要处理接下来 $n$ 天的借教室信息，其中第 $i$ 天学校有 $r_i$ 个教室可供租借。共有 $m$ 份订单，每份订单用三个正整数描述，分别为 $d_j,s_j,t_j$ ，表示某租借者需要从第 $s_j$ 天到第 $t_j$ 天租借教室（包括第 $s_j$ 天和第 $t_j$ 天），每天需要租借 $d_j$ 个教室。

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

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

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

输入格式

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

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

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

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

输出格式

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

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

样例

样例输入

样例输出

-1 
2

提示

【输入输出样例说明】

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

【数据范围】

对于10%的数据，有 $1 \leq n, m \leq 10$ ；

对于30%的数据，有 $1 \leq n, m \leq 1000$ ；

对于 70%的数据，有 $1 ≤ n,m ≤ 10^5$ ；

对于 100%的数据，有 $1 ≤ n,m ≤ 10^6,0 ≤ r_i,d_j≤ 10^9,1 ≤ s_j≤ t_j≤ n$ 。

NOIP 2012 提高组第二天第二题

一开始思路

看到这道题很容易就能想到暴力做法：开一个数组，每天需要的就从中减去，直到第一个出现小于零（即房间不够用）的情况，跳出循环输出这一天

代码（55分）

#include <bits/stdc++.h>
#define int long long
#define endl '\n'
using namespace std;
int n, m, i, j, r[1000005], d, s, t, flg;
signed main() {
    ios::sync_with_stdio(0);cin.tie(0);cout.tie(0);
    cin >> n >> m;
    for (i = 1; i <= n; i++)
        cin >> r[i];
    for (j = 1; j <= m; j++) {
        cin >> d >> s >> t;
        for (i = s; i <= t; i++) {
            r[i] -= d;
            if (r[i] < 0) {
                flg = 1;
                goto end;
            }
        }
    }
    end:
    if (flg == 1)  cout << "-1" << endl << j;
    else cout << "0";
    return 0;
}

如果代码这么容易想，肯定不是黄题，故喜提55分

分析问题

这道题的标签写了“二分、前缀和、差分”

1.暴力的时间复杂度为 $n^2$ ，时间主要花在开一个for循环来减去a[]数组中每一天的房间，那么有什么办法能够避免这种操作吗？——前缀和&差分应运而生

从区间[ l , r ]中增加一个数n，那么对于原数组a需要用迭代器 i 从 l 遍历到 r ，但如果我们开一个差分数组d[ ]，那么对于这个操作只需d[l]+=n,d[r+1]-=n，还原a数组只需做一次累加即可

2.对于每日的房间这样单调的增长，使用二分查找以节省时间，不必从头到尾找是哪个房间导致的超容

AC代码

#include <bits/stdc++.h>
#define int long long
#define endl '\n'
#define N 1000005
using namespace std;
int n, m, r[N], d[N], s[N], t[N], diff[N], need[N];
bool judge(int x) {
    memset(diff, 0, sizeof(diff));  //每次都要一个新的diff，否则需要重置操作
    for (int i = 1; i <= x; i++) {	//差分操作
        diff[s[i]] += d[i];
        diff[t[i] + 1] -= d[i];
    }
    for (int i = 1; i <= n; i++) {	//前缀和复原
        need[i] = need[i - 1] + diff[i];
        if (need[i] > r[i])    return 0;
    }
    return 1;
}
signed main() {
    ios::sync_with_stdio(0);cin.tie(0);cout.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 (judge(m))   cout << "0";
    else {
        int begin = 1, end = m;
        while (begin < end)
        {
            int mid = (begin + end) / 2;
            if (judge(mid))begin = mid + 1;
            else end = mid;
        }
        cout << "-1" << endl << begin;
    }
    return 0;
}