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;
}