刷题记录|[NOIP2012]借教室题解

题目描述

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

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

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

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

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

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

输入描述

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

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

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

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

输出描述

如果所有订单均可满足，则输出只有一行，包含一个整数0。否则（订单无法完全满足）输出两行，第一行输出一个负整数-1，第二行输出需要修改订单的申请人编号。

输入样例

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

输出样例

-1
2

解题思路：二分+差分

用二分的思路，假设mid为要修改的人的前一个人，然后判断这mid是否符合条件，若符合则l=mid+1，否则r=mid-1，在判断mid是否符合条件时，由于他在sj到tj他们要借教室，从sj到tj这个区间的教室要相应减少，这时候就想到用差分数组delta维护。由于r数组装的是每天所有的教室个数，所以delta[i] = r[i] - r[i - 1]，由于mid为要修改的人的前一个人，所以从1到mid的订单都满足，所以delta数组的相应区间要进行减少，操作完之后，对差分数组求和，判断得到的结果的正负，正就为真，负就为假。在输出时，如果所有订单都满足的话，l=mid+1,此时就输出0，否则就输出l。

AC代码

#include<bits/stdc++.h>
using namespace std;
int r[1000100];
int n, m;
struct ty
{
    int d;
    int s;
    int t;
}a[1000100];
int delta[1000100];//差分数组,记录当天与前一天ri的差值

int judge(int mid)
{
    memset(delta, 0, sizeof(delta));
    for (int i = 1; i <= n; i++)
    {
        delta[i] = r[i] - r[i - 1];
    }
    for (int i = 1; i <= mid; i++)
    {
        delta[a[i].s] -= a[i].d;
        delta[a[i].t + 1] += a[i].d;
    }
    int sum = 0;
    for (int i = 1; i <=n; i++)//差分数组求和
    {
        sum += delta[i];
        if (sum < 0)
            return 0;
    }
    return 1;
}
int main()
{
    scanf("%d%d", &n, &m);//n, m，表示天数和订单的数量。
    for (int i = 1; i <= n; i++)
        scanf("%d", &r[i]);
    for (int i = 1; i <= m; i++)
        scanf("%d%d%d", &a[i].d, &a[i].s, &a[i].t);
    int l = 0, r = m;//右边界是最后一个订单
    while (l <= r)
    {
        int mid = (l + r) >> 1;//mid是假设的需要修改订单的前一个人
        if (judge(mid))
            l = mid + 1;
        else
            r = mid - 1;
    }
    if (l > m)cout << 0;
    else cout << "-1\n" << l;

    return 0;
}