LeetCode 1296. 划分数组为连续数字的集合
LeetCode 1296. 划分数组为连续数字的集合
题目来源：
来源：力扣（LeetCode）
链接：https://leetcode-cn.com/problems/divide-array-in-sets-of-k-consecutive-numbers
著作权归领扣网络所有。商业转载请联系官方授权，非商业转载请注明出处。
题目描述：
给你一个整数数组 nums 和一个正整数 k，请你判断是否可以把这个数组划分成一些由 k 个连续数字组成的集合。
如果可以，请返回 True；否则，返回 False。
示例1：
输入：nums = [1,2,3,3,4,4,5,6], k = 4 输出：true 解释：数组可以分成 [1,2,3,4] 和 [3,4,5,6]。
示例2：
输入：nums = [3,2,1,2,3,4,3,4,5,9,10,11], k = 3 输出：true 解释：数组可以分成 [1,2,3] , [2,3,4] , [3,4,5] 和 [9,10,11]。
示例 3：
输入：nums = [3,3,2,2,1,1], k = 3 输出：true
示例 4：
输入：nums = [1,2,3,4], k = 3 输出：false 解释：数组不能分成几个大小为 3 的子数组。
提示：
1 <= nums.length <= 10^5 1 <= nums[i] <= 10^9 1 <= k <= nums.length
思路：
根据提示中给出的数据范围来看，数据量并不是很大，数组最大长度为10^5，可以考虑先对数组进行排序，之后通过模拟的方式来判断是否符合题目描述。
模拟方法如下：
1、首先对数组进行排序，并计算出每一个数字出现的个数，推荐使用TreeMap，key存储数字，value存储数字出现的个数。
2、对排好序的数组从第1位遍历到第k位，检查除第1位以外，是否每一位都满足：
①第i位数字是否存在
②第i位数字 = 第i-1位数字+1
③第i位数字出现的个数 >= 第1位数字出现的个数
如果出现任意一次不满足的情况，返回false即可。
3、修改 第i位数字出现的个数 为 第i位数字出现的个数 - 第1位数字出现的个数，如果此时第i位数字出现的个数是0，则删除这个数字。
4、重复进行这个过程直到数组为空。
代码如下：
public boolean isPossibleDivide(int[] nums, int k) {
TreeMap<Integer, Integer> map = new TreeMap<>();
// 对数组进行遍历，加入的TreeMap中，完成了排序及计数
for (int num : nums) {
if (!map.containsKey(num)) {
map.put(num, 1);
} else {
map.put(num, map.get(num) + 1);
}
}
while (map.size() != 0) {
// 取出第1个元素，并将第一个元素删除
Map.Entry<Integer, Integer> start = map.firstEntry();
map.remove(start.getKey());
Iterator<Map.Entry<Integer, Integer>> iterator = map.entrySet().iterator();
// 对数组进行遍历，因为已经取出了一个元素，所以是从第一位到k-1位，而不是k位。
for (int i = 1; i < k; i++) {
// 如果不存在下一个元素，直接返回false，条件①
if (!iterator.hasNext()) {
return false;
}
Map.Entry<Integer, Integer> tmp = iterator.next();
// 条件②、条件③
if (tmp.getKey() != start.getKey() + i || tmp.getValue() < start.getValue()) {
return false;
}
// 步骤3，减少数字的个数，如果数量是0，删除这个数字
if (tmp.getValue().equals(start.getValue())) {
iterator.remove();
} else {
tmp.setValue(tmp.getValue() - start.getValue());
}
}
}
return true;
}