noip2019普及组 公交换乘

题目

公交换乘

算法

(滑动窗口,队列) $O(45n)$

这道题目有个限制很重要：

不会有两次乘车记录出现在同一分钟

这就说明对于每个公交车来说，所能使用的优惠券不会超过45张。即我们可以完美避开各种复杂的数据结构优化了。

因此我们可以从前往后扫描每条记录，同时用一个队列维护当前车次可以使用的优惠券区间：

• 如果当前记录是火车，则加入维护的优惠券区间；
• 如果当前记录是公交车，则线性扫描一遍队列中所有优惠券，找到第一个未被使用过的且大于等于当前价格的优惠券即可；
• 可以用一个bool数组对优惠券判重，以保证每张优惠券最多只被用一次。

时间复杂度

最坏情况下，对于每条公交车记录都需要遍历45张优惠券，因此时间复杂度是 $O(45n)$。

C++ 代码

#include <cstdio>
#include <cstring>
#include <iostream>
#include <algorithm>

using namespace std;

const int N = 100010;

int n;
//优惠券
struct Ticket
{
    int time, price;
    bool used;
} tk[N];

int main()
{
    scanf("%d", &n); // 表示总记录数量

    int res = 0; // 总花费
    for (int i = 0, l = 0, r = 0; i < n; i++)
    {
        int type, price, time;
        scanf("%d%d%d", &type, &price, &time);
        if (type == 0)
        {
            res += price;
            tk[r++] = {time, price};
        }
        else
        {
            while (l < r && time - tk[l].time > 45) l++; //弹出不在窗口内的所有优惠券

            bool success = false;
            for (int i = l; i < r; i++)
                if (tk[i].used == false && tk[i].price >= price)
                {
                    tk[i].used = true; //表示优惠券被用过了，防止重复使用
                    success = true;
                    break;
                }
            if (!success)
                res += price;
        }
    }

    printf("%d\n", res);

    return 0;
}