LeetCode 第 203 场周赛 题解及思路

赛题传送门

1560. 圆形赛道上经过次数最多的扇区

赛题

给你一个整数 n 和一个整数数组 rounds。有一条圆形赛道由 n 个扇区组成，扇区编号从 1 到 n 。现将在这条赛道上举办一场马拉松比赛，该马拉松全程由 m 个阶段组成。其中，第 i 个阶段将会从扇区 rounds[i - 1] 开始，到扇区 rounds[i] 结束。举例来说，第 1 阶段从 rounds[0] 开始，到 rounds[1] 结束。
请你以数组形式返回经过次数最多的那几个扇区，按扇区编号 升序 排列。
注意，赛道按扇区编号升序逆时针形成一个圆。

简单模拟即可，对于每个 round 我们将其经过的扇区对应加一，用哈希表计数，注意边界值处理。

int start = rounds[0] == 1 ? n : rounds[0] - 1;
unordered_map<int, int> counts;
for (int i = 0; i < rounds.size(); i++) {
	do {
		start = (start + 1);
		if (start > n) start = 1;
		counts[start]++;
	} while (start != rounds[i]);
};

然后遍历哈希表找出次数最多的那几个扇区。

vector<int> ans;
int tot = 0;
for (auto iter : counts) {
	if (ans.empty() || iter.second == tot) {
		ans.push_back(iter.first);
		tot = iter.second;
	} else if (iter.second > tot) {
		ans.clear();
		ans.push_back(iter.first);
		tot = iter.second;
	} else {};
};

最后别忘记按照扇区编号升序排序。

sort(ans.begin(), ans.end());

完整代码见 GitHub。

1561. 你可以获得的最大硬币数目

赛题

有 3n 堆数目不一的硬币，你和你的朋友们打算按以下方式分硬币：每一轮中，你将会选出 任意 3 堆硬币（不一定连续）。Alice 将会取走硬币数量最多的那一堆。你将会取走硬币数量第二多的那一堆。Bob 将会取走最后一堆。重复这个过程，直到没有更多硬币。
给你一个整数数组 piles ，其中 piles[i] 是第 i 堆中硬币的数目。返回你可以获得的最大硬币数目。

这是个策略问题，相当于每轮中，你都只能拿到第二大的硬币，所以每次都不得不牺牲一个最大的硬币，但是另外一堆硬币你却可以有所选择，当然是选择剥削别人让他们拿竟可能的小的硬币，这样在之后的轮中你就可以拿到相对大一点的了。

弱肉强食便是这个道理，你打不过的当然是乖乖认怂，打不过你的当然就狠狠剥削他们，来满足自身的需求和贪婪。

回到这道题就是，每轮都选取可选择的第二大的硬币，然后去掉可选择的第一大、第二大和最小的硬币，并进入到下一轮。因为是 3n 个硬币，因此你必然是拿 n 个，所以可以直接遍历数组进行模拟。

当然，为了能快速选择到大的硬币，首先要对原数组进行 降序 排序。

sort(piles.begin(), piles.end(), greater<int>());

然后每轮选取第二大的，一共进行 n 轮结束。

int tot = 0, n = piles.size() / 3;
for (int i = 1; n-- > 0; i += 2)
	tot += piles[i];
return tot;

完整代码见 GitHub。

1562. 查找大小为 M 的最新分组

赛题传送门

给你一个数组 arr ，该数组表示一个从 1 到 n 的数字排列。有一个长度为 n 的二进制字符串，该字符串上的所有位最初都设置为 0 。
在从 1 到 n 的每个步骤 i 中（假设二进制字符串和 arr 都是从 1 开始索引的情况下），二进制字符串上位于位置 arr[i] 的位将会设为 1 。
给你一个整数 m ，请你找出二进制字符串上存在长度为 m 的一组 1 的最后步骤。一组 1 是一个连续的、由 1 组成的子串，且左右两边不再有可以延伸的 1 。
返回存在长度 恰好 为 m 的 一组 1 的最后步骤。如果不存在这样的步骤，请返回 -1 。

想过很多方法，比如可以逆向思维，先将所有该设置为 1 的位先设置了，然后再倒着回来将每个位置为 0 并判断有没有长度为 m 的一组 1，并且用前缀和以及哈希表来记录数组当前的连续 1 的长度，但很快就发现每一次置为 0 都是对连续 1 的破坏，相当于将一个长度为 m1 的连续 1 拆分成两个长度分别为 m2 和 m3 的连续 1，且 m2 + m3 = m1 - 1，而前缀和也会因此招到破坏而需要重新求解，因此没有办法达到 $O(1)$ 复杂度来进行每次操作，所以这样是行不通的。

因此，脑袋一热，想到了这不就是连通性问题嘛，并查集 走起。我们将数组下标看做一个个元素，最开始每个元素都不连通，且 秩 为 0。然后顺序遍历 arr，对于每次操作 i，首先将对应的元素的 秩 设置为 1，然后再 试图连通 arr[i] 和 arr[i - 1]，以及 arr[i] 和 arr[i + 1]，试图 代表着当且仅当两个元素的 秩 都大于 0 才能进行，然后判断有没有元素的 秩 为 m 有的话即可更新最后步骤为 i，这里可以用 哈希表 记录所有元素的 秩 出现次数，方便 $O(1)$ 判断。

具体实现的话，在主函数中，我们先建立一个并查集，并将数组长度以及 m 传入作为参数。然后对于每个元素 arr[i]，我们首先设置他的秩为 1，然后再试图连通他与周围两个元素，操作完成后判断是否存在秩为 m 的元素并更新结果。

Union un(arr.size(), m);
int ans = -1;
for (int i = 0; i < arr.size(); i++) {
	un.setrank(arr[i]);
	un.connect(arr[i], arr[i] - 1);
	un.connect(arr[i], arr[i] + 1);
	if (un.OK()) ans = i + 1;
};
</