【C++堆(优先队列)】1942. 最小未被占据椅子的编号|1695

闻缺陷则喜何志丹

于 2024-10-08 17:00:00 发布

阅读量945

点赞数 22

文章标签： c++ 算法力扣堆优先队列最小未被占据

本文链接：https://blog.csdn.net/he_zhidan/article/details/142286463

版权

工作级难度算法题解专栏收录该内容

189 篇文章 0 订阅

订阅专栏

本文涉及知识点

C++堆(优先队列)

LeetCode1942. 最小未被占据椅子的编号

有 n 个朋友在举办一个派对，这些朋友从 0 到 n - 1 编号。派对里有无数张椅子，编号为 0 到 infinity 。当一个朋友到达派对时，他会占据编号最小且未被占据的椅子。
比方说，当一个朋友到达时，如果椅子 0 ，1 和 5 被占据了，那么他会占据 2 号椅子。
当一个朋友离开派对时，他的椅子会立刻变成未占据状态。如果同一时刻有另一个朋友到达，可以立即占据这张椅子。
给你一个下标从 0 开始的二维整数数组 times ，其中 times[i] = [arrivali, leavingi] 表示第 i 个朋友到达和离开的时刻，同时给你一个整数 targetFriend 。所有到达时间互不相同。
请你返回编号为 targetFriend 的朋友占据的椅子编号。
示例 1：
输入：times = [[1,4],[2,3],[4,6]], targetFriend = 1
输出：1
解释：

朋友 0 时刻 1 到达，占据椅子 0 。
朋友 1 时刻 2 到达，占据椅子 1 。
朋友 1 时刻 3 离开，椅子 1 变成未占据。
朋友 0 时刻 4 离开，椅子 0 变成未占据。
朋友 2 时刻 4 到达，占据椅子 0 。
朋友 1 占据椅子 1 ，所以返回 1 。
示例 2：
输入：times = [[3,10],[1,5],[2,6]], targetFriend = 0
输出：2
解释：
朋友 1 时刻 1 到达，占据椅子 0 。
朋友 2 时刻 2 到达，占据椅子 1 。
朋友 0 时刻 3 到达，占据椅子 2 。
朋友 1 时刻 5 离开，椅子 0 变成未占据。
朋友 2 时刻 6 离开，椅子 1 变成未占据。
朋友 0 时刻 10 离开，椅子 2 变成未占据。
朋友 0 占据椅子 2 ，所以返回 2 。
提示：
n == times.length
2 <= n <= 10⁴
times[i].length == 2
1 <= arrivali < leavingi <= 10⁵
0 <= targetFriend <= n - 1
每个 arrivali 时刻互不相同。

堆（优先队列)

小根堆记录空闲的椅子，初始为[0,n-1)。
小根堆记录离开的情况，每个元素包括：{离开时间，座位号}。
一，将times的下标indexs按到达时间排序。
二，i:indexs枚举times[i]：
a,将离开时间小于等于times[i][0]的椅子释放。
b，为i分配座位。
c,如果targetFriend是i返回分配的座位。

代码

核心代码

class Solution {
		public:
			int smallestChair(vector<vector<int>>& times, int targetFriend) {
				const int N = times.size();
				vector<int> indexs(N);
				iota(indexs.begin(), indexs.end(), 0);
				sort(indexs.begin(), indexs.end(), [&](const int& i1, const int& i2) {return times[i1][0] < times[i2][0]; });
				priority_queue<int, vector<int>, greater<>> can ;
				for (int i = 0; i < N; i++) {
					can.emplace(i);
				}
				priority_queue<pair<int,int>, vector<pair<int, int>>, greater<pair<int, int>>> heapSeat;
				for (const auto& i : indexs) {
					while (heapSeat.size() && (heapSeat.top().first <= times[i][0])) {
						can.emplace(heapSeat.top().second);
						heapSeat.pop();
					}
					if (i == targetFriend) { return can.top(); }
					heapSeat.emplace(times[i][1], can.top());
					can.pop();
				}
				return -1;
			}
		};

单元测试

vector<vector<int>> times;
		int targetFriend;
		TEST_METHOD(TestMethod11)
		{
			times = { {1,4},{2,3},{4,6} }, targetFriend = 1;
			auto res = Solution().smallestChair(times, targetFriend);
			AssertEx(1, res);
		}
		TEST_METHOD(TestMethod12)
		{
			times = { {3,10},{1,5},{2,6} }, targetFriend = 0;
			auto res = Solution().smallestChair(times, targetFriend);
			AssertEx(2, res);
		}

扩展阅读

我想对大家说的话
工作中遇到的问题，可以按类别查阅鄙人的算法文章，请点击《算法与数据汇总》。
学习算法：按章节学习《喜缺全书算法册》，大量的题目和测试用例，打包下载。重视操作
有效学习：明确的目标及时的反馈拉伸区（难度合适）专注
闻缺陷则喜(喜缺)是一个美好的愿望，早发现问题，早修改问题，给老板节约钱。
子墨子言之：事无终始，无务多业。也就是我们常说的专业的人做专业的事。
如果程序是一条龙，那算法就是他的是睛
失败+反思=成功成功+反思=成功