LeetCode-78 Subsets

it2023-09-03  77

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
最新回复(0)