宝宝也能看懂的 leetcode 周赛 - 168 - 2
Hi 大家好，我是张小猪。欢迎来到『宝宝也能看懂』系列之 leetcode 题解。
这里是第 168 期的第 2 题，也是题目列表中的第 1296 题 -- 『Divide Array in Sets of K Consecutive Numbers』
题目描述
Given an array of integers nums and a positive integer k, find whether it's possible to divide this array into sets of k consecutive numbers
Return True if its possible otherwise return False.
Example 1:
Input: nums = [1,2,3,3,4,4,5,6], k = 4
Output: true
Explanation: Array can be divided into [1,2,3,4] and [3,4,5,6].Example 2:
Input: nums = [3,2,1,2,3,4,3,4,5,9,10,11], k = 3
Output: true
Explanation: Array can be divided into [1,2,3] , [2,3,4] , [3,4,5] and [9,10,11].Example 3:
Input: nums = [3,3,2,2,1,1], k = 3
Output: trueExample 4:
Input: nums = [1,2,3,4], k = 3
Output: false
Explanation: Each array should be divided in subarrays of size 3.Constraints:
1 <= nums.length <= 10^5
1 <= nums[i] <= 10^9
1 <= k <= nums.length官方难度
MEDIUM
解决思路
题目内容不长，是对于一个条件的判断。即尝试把输入数组进行拆分，拆分后每一段的长度都要为 k，并且每一段都要是连续的数字。
看完题目之后，第一反应，数组长度可以先做判断，不能整除 k 就一定无法满足。接下来思路的大体方向就是，从最小或者最大开始，逐渐扫出符合要求的子段，然后剔除出去，直到遇到不符合要求的情况，或者剩下的内容为空为止。
直接方案
整体思路感觉可行后，便开始具体施工。为了从一端开始，我们首先对数组进行了排序。接下来最关键的就是对子段的剔除操作。这里可能有两种常见的方式：
- 第一种方式，即是真正的执行这个剔除操作。这样做的话会受到底层数据结构的影响，例如如果是线性表，那么在中间进行部分移除操作会消耗蛮多的代价；而如果是链表，那么频繁的查询又是一个问题。而 JS 语言原生其实并没有太多的内置数据结构。
- 第二种方式，即用计数法来处理剔除行为。这种方式我们不需要进行实际的剔除行为，只需要在 O(1) 的时间索引到目标，然后改变计数值即可。不过代价是需要 O(n) 的时间来初始化计数，以及占用额外的空间进行储存。
考虑到希望整体时间能更加快，所以这里采用了第二种方式。由于数字的范围较大，[1, 10^9]，所以使用了 Map 而不是固定长度的数组来储存计数。
const isPossibleDivide = (nums, k) => {
if (nums.length % k !== 0) return false;
const map = new Map();
nums.sort((a, b) => a - b);
for (const val of nums) {
map.set(val, map.has(val) ? map.get(val) + 1 : 1);
}
for (let i = 0; i < nums.length; ++i) {
const start = nums[i];
const count = map.get(start);
if (count === 0) continue;
for (let i = 1; i < k; ++i) {
const c = map.get(start + i)
if (c === undefined || c < count) return false;
map.set(start + i, c - count);
}
map.set(start, 0);
}
return true;
};试了一下，Accepted，152ms。
优化
回头看代码，这个针对全数组的排序一直很让我在意，因为单看它就会承担着比较大的额外性能负担，而它的收益仅仅是确保了我们从数据的一端开始进行剔除操作。那么我们是否有其他办法保证这件事情呢？
一个非常简单的思路是，对于我们遍历到的数字，我们尝试找到它前面最近的一个断点，这样我们便满足了需求。并且随着一次次的剔除操作，这个查找断点的过程会更加的简单。
const isPossibleDivide = (nums, k) => {
if (nums.length % k !== 0) return false;
const map = new Map();
for (const val of nums) {
map.set(val, map.has(val) ? map.get(val) + 1 : 1);
}
for (let start of nums) {
if (map.get(start) === 0) continue;
while (map.get(--start) > 0);
++start;
const count = map.get(start);
for (let i = 1; i < k; ++i) {
const c = map.get(start + i)
if (c === undefined || c < count) return false;
map.set(start + i, c - count);
}
map.set(start, 0);
}
return true;
};这段代码最终跑到了 92ms，暂时 beats 100%。于是先凑合着这样了。
相关链接
