leetcode-491. 递增子序列

题目

给定一个整型数组, 你的任务是找到所有该数组的递增子序列，递增子序列的长度至少是2。

示例:

输入: [4, 6, 7, 7]
输出: [[4, 6], [4, 7], [4, 6, 7], [4, 6, 7, 7], [6, 7], [6, 7, 7], [7,7], [4,7,7]]

说明:

给定数组的长度不会超过15。
数组中的整数范围是 [-100,100]。
给定数组中可能包含重复数字，相等的数字应该被视为递增的一种情况。

解题思路

和最长上升子序列比较相似，用dp[i]来表示以nums[i]结尾的数组的递增子序列们，则最终答案就是dp中所有符合要求的子序列

时间复杂度$o(n^2)$

比较麻烦的一点是需要对最终答案进行去重，这里去重的方法是，将序列list转换成tuple，然后再用set

代码

class Solution:
    def findSubsequences(self, nums: List[int]) -> List[List[int]]:
        dp = [[[each_num]] for each_num in nums]
        for i in range(len(nums)):
            for prev in range(i):
                if nums[i] >= nums[prev]:
                    dp[i] += [item + [nums[i]] for item in dp[prev]]
        ans = set()
        for each_dpi in dp:
            ans.update([tuple(item) for item in each_dpi if len(item) >= 2])
        return [list(item) for item in ans]

看到了一个别人的题解，比我自己的写法简洁很多：

class Solution:
    def findSubsequences(self, nums: List[int]) -> List[List[int]]:
        if not nums:
            return []
        dp = {(nums[0], )}
        for each_num in nums[1:]:
            dp.update({prev + (each_num, ) for prev in dp if prev[-1] <= each_num})
            dp.add((each_num, ))
        return [list(item) for item in dp if len(item) >= 2]