LeetCode-78 Subsets

Description

Given a set of distinct integers, nums, return all possible subsets (the power set).

Note: The solution set must not contain duplicate subsets.

Example

Example:
Input: nums = [1,2,3]
Output:
[
[3],
[1],
[2],
[1,2,3],
[1,3],
[2,3],
[1,2],
[]
]

Submissions

解题思路主要在上一题的解题思路上进行了修改，由于上一题给定了子集长度k，因此这题在递归函数时只需添加循环，遍历子集长度为0到len(nums)即可。
上一题思路采用递归，先选定一个数，例如[1]，再依次往后顺序添加，例如[1, 2]，长度等于k时，即将tmp列表添加到res，然后return，后tmp列表抛出最后一位，例如2，加入下一位，例如3，形成[1, 3]，依次递归。且判断当后面数字的个数小于当前tmp所需数字个数时，可直接跳出递归，即直接结束本次循环。
实现代码如下：

class Solution:
    def subsets(self, nums: List[int]) -> List[List[int]]:
        res = []
        def dfs(start, tmp, k):
            if len(tmp) == k:
                res.append(list(tmp))
                return
            for i in range(start, len(nums)):
                if k - len(tmp) - 1 > (len(nums) - i):
                    break
                tmp.append(nums[i])
                dfs(i + 1, tmp, k)
                tmp.pop()
        for j in range(len(nums)+1):
            dfs(0, [], j)
        return res