题解：洛谷 P1083 [NOIP 2012 提高组] 借教室

【题目来源】

洛谷：[P1083 NOIP 2012 提高组] 借教室 - 洛谷

【题目描述】

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

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

我们需要处理接下来 \(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\)，第二行输出需要修改订单的申请人编号。

【输入样例】

4 3 
2 5 4 3 
2 1 3 
3 2 4 
4 2 4

【输出样例】

-1 
2

【算法标签】

《洛谷 P1083 借教室》 #二分# #前缀和# #差分# #NOIP提高组# #2012#

【代码详解】

#include <bits/stdc++.h>
using namespace std;

#define int long long  // 定义int为long long类型

// 全局变量声明
int n, m;               // n: 天数，m: 订单数
int r[1000005];         // 每天可用的教室数量
int d[1000005];         // 每个订单需要的教室数
int s[1000005], t[1000005]; // 每个订单的起始和结束天数
int ans;                // 第一个无法满足的订单编号
int num[1000005];       // 差分数组，用于计算每天教室使用量

/**
 * 检查前x个订单是否能被满足
 * @param x 要检查的订单数量
 * @return 是否能满足所有订单需求
 */
bool check(int x)
{
    memset(num, 0, sizeof(num));  // 初始化差分数组
    
    // 处理前x个订单，构建差分数组
    for (int i = 1; i <= x; i++)
    {
        num[s[i]] += d[i];        // 起始天增加需求
        num[t[i] + 1] -= d[i];   // 结束天+1减少需求
    }
    
    // 计算前缀和，得到每天的实际使用量
    for (int i = 1; i <= n; i++)
    {
        num[i] = num[i - 1] + num[i];  // 计算前缀和
        if (num[i] > r[i])             // 如果超过可用教室数
            return false;
    }
    return true;
}

/**
 * 二分查找第一个无法满足的订单
 * @return 最后一个能满足的订单编号
 */
int find()
{
    int l = 0, r = m + 1;  // 初始化二分边界
    
    // 二分查找过程
    while (l + 1 < r)
    {
        int mid = (l + r) / 2;
        if (check(mid))     // 如果能满足前mid个订单
            l = mid;        // 尝试更大的订单数
        else
            r = mid;        // 否则尝试更小的订单数
    }
    return l;
}

signed main()
{
    // 输入数据
    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];  // 订单信息
    }
    
    // 查找第一个无法满足的订单
    ans = find() + 1;
    
    // 处理输出结果
    if (ans == m + 1)  // 如果所有订单都能满足
    {
        cout << 0;
    }
    else               // 否则输出第一个无法满足的订单
    {
        cout << -1 << endl << ans << endl;
    }
    
    return 0;
}

#include <bits/stdc++.h>
using namespace std;

#define int long long  // 定义int为long long类型
const int N = 1000006;  // 定义数组最大长度

// 全局变量声明
int n, m;               // n: 天数，m: 订单数
int r[N];               // 每天可用的教室数量
int d[N], s[N], t[N];   // 订单信息：d[i]需求数，s[i]起始天，t[i]结束天
int ans;                // 第一个无法满足的订单编号
int num[N];             // 差分数组，用于计算每天教室使用量

/**
 * 检查前x个订单是否能被满足
 * @param x 要检查的订单数量
 * @return 是否能满足所有订单需求
 */
bool check(int x)
{
    memset(num, 0, sizeof(num));  // 初始化差分数组
    
    // 处理前x个订单，构建差分数组
    for (int i = 1; i <= x; i++)
    {
        num[s[i]] += d[i];        // 起始天增加需求
        num[t[i] + 1] -= d[i];    // 结束天+1减少需求
    }
    
    // 计算前缀和，得到每天的实际使用量
    for (int i = 1; i <= n; i++)
    {
        num[i] = num[i - 1] + num[i];  // 计算前缀和
        if (num[i] > r[i])             // 如果超过可用教室数
            return false;
    }
    return true;
}

/**
 * 二分查找第一个无法满足的订单
 * @return 最后一个能满足的订单编号
 */
int find()
{
    int l = 0, r = m;  // 初始化二分边界
    
    // 二分查找过程
    while (l < r)
    {
        int mid = (l + r + 1) >> 1;  // 向上取整
        if (check(mid))              // 如果能满足前mid个订单
            l = mid;                 // 尝试更大的订单数
        else
            r = mid - 1;             // 否则尝试更小的订单数
    }
    return l;
}

signed main()
{
    // 输入数据
    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];  // 订单信息
    
    // 查找第一个无法满足的订单
    ans = find() + 1;
    
    // 处理输出结果
    if (ans == m + 1)  // 如果所有订单都能满足
    {
        cout << 0;
    }
    else                // 否则输出第一个无法满足的订单
    {
        cout << -1 << endl << ans << endl;
    }
    
    return 0;
}

【运行结果】

4 3
2 5 4 3
2 1 3
3 2 4
4 2 4
-1
2