用Python3解决LeetCode子集（迭代+回溯）题目迭代法回溯算法

148 阅读 0 评论 98 点赞

我是靠谱客的博主负责砖头，这篇文章主要介绍用Python3解决LeetCode子集（迭代+回溯）题目迭代法回溯算法，现在分享给大家，希望可以做个参考。

题目

给定一组不含重复元素的整数数组 nums，返回该数组所有可能的子集（幂集）。

说明：解集不能包含重复的子集。

示例:

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

来源：力扣（LeetCode）
链接：https://leetcode-cn.com/problems/subsets
著作权归领扣网络所有。商业转载请联系官方授权，非商业转载请注明出处。

迭代法

代码块

复制代码

1
2
3
4
5
6
7
8
9
class Solution:
    def subsets(self, nums: List[int]) -> List[List[int]]:
        res = [[]]
        n = len(nums)
        for i in nums:
            for j in res:
                res = res + [[i] + j]
        return res

两个循环，进行遍历+迭代，不断更新结果List，最后返回res。

运行结果

在这里插入图片描述

回溯算法

代码块

复制代码

1
2
3
4
5
6
7
8
9
10
11
class Solution:
    def subsets(self, nums: List[int]) -> List[List[int]]:
        res = []
        n = len(nums)
        def hs(i, temp):
            res.append(temp)
            for j in range(i, n):
                hs(j + 1, temp + [nums[j]])
        hs(0, [])
        return res