算法设计与分析—子集(回溯法)

算法设计与分析课程练习:

78. 子集
给你一个整数数组 nums ,数组中的元素 互不相同 。返回该数组所有可能的子集(幂集)。
解集 不能 包含重复的子集。你可以按 任意顺序 返回解集。

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

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

来源:力扣(LeetCode)

a) 算法思路
1、采用组合数学经常使用的思路,放或不放
2、比如1、2、3三个格子,从1开始,有两种选择,1选或者不选,然后再判断2,选或者不选。
3、两种选择分别进行递归
下面的回溯算法可以作为回溯的模板(感谢刘安老师提供的模板)

def subsets(self, nums):
    """
    :type nums: List[int]
    :rtype: List[List[int]]
    """
    n=len(nums)
    res=[]
    path=[]
    def generate_subset(i):
        if i>=n:
            res.append(path[:])
        else:
            for c in range(2):  #重点选或者不选
                if c==0:  #c==0表示选中
                    path.append(nums[i])
                    generate_subset(i+1)   # 选要递归
                    path.pop()
                else:
                    generate_subset(i+1)   # 不选也要递归
    generate_subset(0)
    return res

猜你喜欢

转载自blog.csdn.net/qq_39740279/article/details/121308948
今日推荐