LeetCode 每日一题659. 分割数组为连续子序列

659. 分割数组为连续子序列

给你一个按升序排序的整数数组 num（可能包含重复数字），请你将它们分割成一个或多个子序列，其中每个子序列都由连续整数组成且长度至少为 3 。

如果可以完成上述分割，则返回 true ；否则，返回 false 。

示例 1：

输入: [1,2,3,3,4,5]
输出: True
解释:
你可以分割出这样两个连续子序列 : 
1, 2, 3
3, 4, 5

示例 2：

输入: [1,2,3,3,4,4,5,5]
输出: True
解释:
你可以分割出这样两个连续子序列 : 
1, 2, 3, 4, 5
3, 4, 5

示例 3：

输入: [1,2,3,4,4,5]
输出: False

提示：

• 输入的数组长度范围为 [1, 10000]

解题思路：

把场景想象成 拼团，当有新人进来，有两种选择：

1. 能加入已有的团，就加入已有的团（加入得有规则）
2. 不能加入，就必须拉两个新人开团，拉不到则算失败

带入代码逻辑则是：

1. 遍历到元素 x ，先找是否有 x - 1 结尾的子序列，有则拼接到该子序列
2. 没有则在数组中寻找 x + 1 和 x + 2 成立新的子序列，找不到则返回 false

思考的重心要从 数字 转到 子序列，即在任一时刻所有子序列都必须是合法的。

细节：

• 数组不适合直接找元素，用 countMap 存每个数字的个数，每用一个对应减一
• 没有必要维护整个子序列，用 tailCountMap 存所有子序列最后一个数字的个数

方法一：贪心算法

public boolean isPossible(int[] nums) {
    // 储存每个数字的个数
    Map<Integer, Integer> countMap = new HashMap<>();
    for (int x : nums) {
        countMap.put(x, countMap.getOrDefault(x, 0) + 1);
    }
    // 储存子数组末尾数字的个数
    Map<Integer, Integer> tailCountMap = new HashMap<>();

    for (int x: nums) {
        int count = countMap.get(x);
        if (count > 0) {
            countMap.put(x, count - 1);
            int preTailCount = tailCountMap.getOrDefault(x - 1, 0);
            if (preTailCount > 0) {
                // 如果有 x - 1 结尾的子序列，则拼接到子序列
                tailCountMap.put(x - 1, preTailCount - 1);
                tailCountMap.put(x, tailCountMap.getOrDefault(x, 0) + 1);
            } else {
                // 没有则找 x + 1，x + 2 成立新序列，找不到则返回false
                int count1 = countMap.getOrDefault(x + 1, 0);
                int count2 = countMap.getOrDefault(x + 2, 0);
                if (count1 > 0 && count2 > 0) {
                    countMap.put(x + 1, count1 - 1);
                    countMap.put(x + 2, count2 - 1);
                    tailCountMap.put(x + 2, tailCountMap.getOrDefault(x + 2, 0) + 1);
                } else {
                    return false;
                }
            }
        }
    }
    return true;
}

执行结果