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

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

---

题目来源：LeetCode（力扣）https://leetcode-cn.com/problems/split-array-into-consecutive-subsequences/

题目

---

给你一个按升序排序的整数数组 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 <= nums.length <= 10000

解题思路

---

思路：贪心

先审题，题目要求在给定安升序排序的整数数组 num 中（可能含重复数字），将其分割为一个或多个长度至少为 3 的子序列。其中子序列必须由连续整数组成。若能按照条件完成切割，那么返回 True，否则返回 False。

由于分割后的子序列必须是连续整数组成的，且子序列限制长度必须至少为 3。那么这里要确定子序列，就必须知道子序列的长度以及子序列的末尾元素。

对于数组中每个元素 num 而言，需要对其进行判断。

• 当存在子序列且以 num-1 结尾时，应该考虑将 num 并入子序列中。
• 若不存在，那么再考虑创建新的子序列。

具体的做法如下：

• 首先声明两个字典，count_map 用以存储每个数字剩余的次数，end_map 用以存储每个数字作为结尾的子序列的数量；
• 初始情况下，数字剩余次数即为出现次数，遍历数组，初始化字典 count_map；
• 再次遍历数组，开始对每个数字 num 进行判断。当 num 在 count_map 中剩余的次数大于 0 时，判断是否存在以 num-1 结尾的子序列。
  • 若存在时，将 num 并入该子序列中。此时注意更新两个字典，由于 num 已被并入子序列中，那么 num 剩余次数应减 1，即是 count_map[num] -= 1；同时子序列长度发生变化且以 num 结尾，那么需要更新以数字结尾的子序列数量，即是 end_map[num-1] -= 1，end_map[num] += 1。
  • 若不存在时，应该考虑创建新的子序列。以 num 为子序列的第一个元素，先判断num 后续的两个连续的数字剩余的次数是否都大于 0：
    • 若都大于 0 时，创建新的子序列。同时注意更新字典，即三个数字在 count_map 字典中对应的次数都应减 1，同时更新 end_map 字典，以 num+2 结尾的子序列数量加 1.
    • 如果后续两个连续数字剩余次数不都大于 0，那么表示无法构建长度至少为 3 的子序列，那么表示无法完成切割，此时直接返回 False。
• 若 num 在 count_map 中剩余次数不大于 0 时，不讨论（已经被并入序列中），跳过。
• 当数组遍历结束没有遇到无法分割确定子序列的情况时，返回 True。

代码实现如下。

class Solution:
    def isPossible(self, nums: List[int]) -> bool:
        count_map = collections.defaultdict(int)
        end_map = collections.defaultdict(int)

        # 统计数字出现次数
        for num in nums:
            count_map[num] += 1

        for num in nums:
            count = count_map[num]
            # 只有数字出现次数大于 0，才有讨论的意义
            # 因为数字可能已经被并入其他子序列
            if count > 0:
                # 数字出现次数大于 0 时，查看是否有 num-1 结尾的子序列
                # 若有并入子序列中，否则考虑重新创建新子序列
                pre_end_count = end_map.get(num-1, 0)
                if pre_end_count > 0:
                    count_map[num] -= 1
                    end_map[num-1] = pre_end_count - 1
                    end_map[num] += 1
                else:
                    # 考虑重建新子序列，不满足条件返回 False
                    count1 = count_map.get(num+1, 0)
                    count2 = count_map.get(num+2, 0)
                    if count1 > 0 and count2 > 0:
                        count_map[num] -= 1
                        count_map[num + 1] -= 1
                        count_map[num + 2] -= 1
                        end_map[num + 2] += 1
                    else:
                        return False
        
        return True

欢迎关注

---

公众号 【书所集录】

---

如有错误，烦请指出，欢迎指点交流。