【洛谷 P8685】[蓝桥杯 2019 省 A] 外卖店优先级 题解(模拟+优先队列+位集合)

125 阅读3分钟

[蓝桥杯 2019 省 A] 外卖店优先级

题目描述

“饱了么”外卖系统中维护着 NN 家外卖店,编号 11NN。每家外卖店都有一个优先级,初始时 (0(0 时刻)优先级都为 00

每经过 11 个时间单位,如果外卖店没有订单,则优先级会减少 11,最低减到 00;而如果外卖店有订单,则优先级不减反加,每有一单优先级加 22

如果某家外卖店某时刻优先级大于 55,则会被系统加入优先缓存中;如果优先级小于等于 33,则会被清除出优先缓存。

给定 TT 时刻以内的 MM 条订单信息,请你计算 TT 时刻时有多少外卖店在优先缓存中。

输入格式

第一行包含 33 个整数 NNMMTT

以下 MM 行每行包含两个整数 tstsidid,表示 tsts 时刻编号 idid 的外卖店收到。

一个订单。

输出格式

输出一个整数代表答案。

样例 #1

样例输入 #1

2 6 6
1 1
5 2
3 1
6 2
2 1
6 2

样例输出 #1

1

提示

样例解释

66 时刻时,11 号店优先级降到 33,被移除出优先缓存;22 号店优先级升到 66

加入优先缓存。所以是有 11 家店 (2(2 号)在优先缓存中。

评测用例规模与约定

对于 80%80\% 的评测用例,1N,M,T100001 \le N,M,T \le 10000

对于所有评测用例,1N,M,T1051 \le N,M,T \le 10^51tsT1 \le ts \le T1idN1 \le id \le N

蓝桥杯 2019 年省赛 A 组 G 题。


思路

首先,定义一个结构体Sorder,包含两个成员变量:ts和id,分别表示订单的时间和外卖店的编号。然后,定义了一个比较函数,用于优先队列的排序,时间靠前的优先。

初始化一些变量,包括n(外卖店的数量)、m(订单的数量)、t(时间单位的数量)、p(用于存储每个外卖店的优先级)、pri(用于标记哪些外卖店在优先缓存中)、vis(用于标记哪些外卖店在当前时间单位有订单)等。

然后,从输入中读取了m条订单信息,将每个订单的时间和外卖店编号加入到优先队列pq1中。

接着,对每一个时间单位,都会执行以下操作:首先,重置vis数组,然后,检查优先队列pq1的顶部元素,如果订单的时间等于当前时间,那么就将该订单的外卖店编号对应的优先级加2,并标记该外卖店在当前时间单位有订单,然后将该订单从优先队列中移除。如果外卖店的优先级大于5,那么就将该外卖店加入到优先缓存中。然后,对所有的外卖店,如果没有订单,那么就将其优先级减1,如果优先级小于等于3并且在优先缓存中,那么就将其从优先缓存中移除。

最后,统计并输出在优先缓存中的外卖店的数量。


AC代码

#include <algorithm>
#include <bitset>
#include <cstring>
#include <iostream>
#include <queue>
#define mp make_pair
#define AUTHOR "HEX9CF"
using namespace std;
using ll = long long;

const int N = 1e6 + 7;
const int INF = 0x3f3f3f3f;
const ll MOD = 1e9 + 7;

struct Sorder {
	int ts, id;
	struct cmp {
		bool operator()(const Sorder &a, const Sorder &b) {
			// 时间靠前的优先
			return a.ts > b.ts;
		}
	};
};

int n, m, t;
int p[N];
bitset<N> pri, vis;
priority_queue<Sorder, vector<Sorder>, Sorder::cmp> pq1;

void dec() {
	for (int i = 1; i <= n; i++) {
		if (p[i] && !vis[i]) {
			p[i]--;
			if(pri[i] && p[i] <= 3) {
				// 移出优先缓存
				pri[i] = 0;
			}
		}
	}
}

int main() {
	ios::sync_with_stdio(0);
	cin.tie(0);
	cout.tie(0);

	memset(p, 0, sizeof(p));
	cin >> n >> m >> t;
	for (int i = 1; i <= m; i++) {
		int ts, id;
		cin >> ts >> id;
		pq1.push({ts, id});
	}

	pri.reset();
	for (int i = 1; i <= t; i++) {
		vis.reset();
		while (pq1.size()) {
			auto f = pq1.top();
			if (f.ts == i) {
				// cout << i << " " << f.id << endl;
				p[f.id] += 2;
				vis[f.id] = 1;
				pq1.pop();
				if (p[f.id] > 5) {
					// 加入优先缓存
					pri[f.id] = 1;
				}
			} else {
				break;
			}
		}
		dec();
	}

	ll ans = 0;
	for (int i = 1; i <= n; i++) {
		if (pri[i]) {
			ans++;
		}
		// cout << p[i] << " " << pri[i] << endl;
	}
	cout << ans << "\n";
	return 0;
}