78. 子集问题(回溯算法,子集)

78. 子集


每日几道leetcode刷刷题!
传送门


题目描述

给你一个整数数组 nums ,数组中的元素 互不相同 。返回该数组所有可能的子集(幂集)。

解集 不能 包含重复的子集。你可以按 任意顺序 返回解集。

示例 1:
输入:nums = [1,2,3]
输出:[[],[1],[2],[1,2],[3],[1,3],[2,3],[1,2,3]]


代码

class Solution:
    def subsets(self, nums: List[int]) -> List[List[int]]:
        result = []
        path = []
        def backtrack(num, start):
            result.append(path[:])
            if start >= len(nums):
                return
            for i in range(start, len(nums)):
                path.append(nums[i])
                backtrack(num, i+1)
                path.pop()
        backtrack(nums, 0)
        return result

90. 子集 II

题目描述

给你一个整数数组 nums ,其中可能包含重复元素,请你返回该数组所有可能的子集(幂集)。

解集 不能 包含重复的子集。返回的解集中,子集可以按 任意顺序 排列。

示例 1:
输入:nums = [1,2,2]
输出:[[],[1],[1,2],[1,2,2],[2],[2,2]]

代码

class Solution:
    def subsetsWithDup(self, nums: List[int]) -> List[List[int]]:
        result = []
        path = []
        used = [False] * len(nums)
        def backtrack(nums, start):
            result.append(path[:]) #子集问题,在每次递归都添加一次
            if start == len(nums):
                return
            for i in range(start, len(nums)):
                if i > 0 and nums[i] == nums[i - 1] and not used[i - 1]:
                    continue
                used[i] = True
                path.append(nums[i])
                backtrack(nums, i+1)
                path.pop()
                used[i] = False
        nums.sort() #排序一下
        backtrack(nums, 0)
        return result

总结

组合问题和分割问题都是收集树的叶⼦节点,⽽⼦集问题是找树的所有节点,不需要剪

你可能感兴趣的:(算法刷题,python,leetcode,算法)