题目链接:https://leetcode.cn/problems/divide-array-in-sets-of-k-consecutive-numbers/description/
题目大意:给出一个数组nums[]
和一个数k
,求nums[]
能否被分成若干个k
个元素的连续的子列。
思路:比较简单,贪心就行,找到当前剩下的元素中最小的v
,然后(如果合法)它必然属于某个子列,那么就找v+1, ..., v+k-1
,这些元素的剩余量都减1即可。如果出现空缺,那么就返回false
。
很显然用哈希表比较合适。不过我开始做时,因为要从小到大遍历剩余元素,就用了map<int, int>
,直接从map
的头开始遍历。虽然通过了,但速度有点慢。看了题解,发现用的是unordered_map<int, int>
,区别就是先把nums[]
排序了一遍,然后对nums[]
进行遍历。这也是OK的,因为排序后,nums[]
中,每个最小的元素都需要被归入一个子列中。这样就节约了时间。
完整代码
class Solution {
public:
bool isPossibleDivide(vector<int>& nums, int k) {
int n = nums.size();
if (n % k)
return false;
sort(nums.begin(), nums.end());
unordered_map<int, int> cnt;
for (auto& num : nums) {
cnt[num]++;
}
for (auto& num : nums) {
if (!cnt.count(num))
continue;
cnt[num]--;
if (cnt[num] == 0)
cnt.erase(num);
for (int i = 1; i < k; i++) {
if (cnt.count(num+i) != 0) {
cnt[num+i]--;
if (cnt[num+i] == 0)
cnt.erase(num+i);
}
else
return false;
}
}
return true;
}
};