其他分享
首页 > 其他分享> > 递归&回溯 子集、子集II、幂集

递归&回溯 子集、子集II、幂集

作者:互联网

78. 子集

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

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

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

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

提示:
1 <= nums.length <= 10
-10 <= nums[i] <= 10
nums 中的所有元素 互不相同

回溯算法解决

class Solution:
    def subsets(self, nums: List[int]) -> List[List[int]]:
        if not nums:
            return []
        res = []
        def back(index,tmp):
            res.append(tmp)
            for i in range(index,len(nums)):
                back(i+1,tmp+[nums[i]])
        back(0,[])
        return res

90. 子集 II

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

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

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

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

提示:
1 <= nums.length <= 10
-10 <= nums[i] <= 10

回溯算法解决
class Solution:
    def subsetsWithDup(self, nums: List[int]) -> List[List[int]]:
        if not nums:
            return []
        res = []
        nums = sorted(nums)
        def back(index, tmp):
            res.append(tmp)
            for i in range(index,len(nums)):
            #去除重复元素
                if i> index and nums[i] == nums[i-1]:
                    continue
                back(i+1,tmp + [nums[i]])
        back(0,[])
        return res

标签:tmp,幂集,nums,res,List,back,II,子集
来源: https://blog.csdn.net/mrjkzhangma/article/details/120091044