leetcode练习——回溯算法(子集)

这篇博客介绍了如何使用Python解决LeetCode上的子集问题。提供了三种解法,包括利用itertools库的combinations函数,通过迭代生成子集,以及使用递归和回溯算法实现。每种方法都具有36ms的运行时间和适度的内存消耗。

给你一个整数数组 nums ,数组中的元素 互不相同 。返回该数组所有可能的子集(幂集)。
解集 不能 包含重复的子集。你可以按 任意顺序 返回解集。
在这里插入图片描述

精彩解法:https://leetcode-cn.com/problems/subsets/solution/hui-su-suan-fa-by-powcai-5/

解法一:库函数(36ms/14.9MB)

  • itertools.combinations()
itertools.combinations(iterable[, r]

创建一个迭代器,返回iterable中所有长度为r的子序列,返回的子序列中的项按输入iterable中的顺序排序 (不带重复).
r 指定生成排列的元素的长度,如果不指定,则默认为可迭代对象的元素长度。
对应组合概念。

>>> from itertools import combinations
>>>
>>> list(combinations('ABC', 2))
[('A', 'B'), ('A', 'C'), ('B', 'C')]
class Solution:
    def subsets(self, nums: List[int]) -> List[List[int]]:
        res = []
        for i in range(len(nums)+1):
            for tmp in itertools.combinations(nums, i):
                res.append(tmp)
        return res

解法二:迭代(36ms/15.2MB)

class Solution:
    def subsets(self, nums: List[int]) -> List[List[int]]:
        res = [[]]
        for i in nums:
            res = res + [[i] + num for num in res]
        return res

解法三:递归(回溯算法)(36ms/15MB)

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

力扣链接:https://leetcode-cn.com/leetbook/read/top-interview-questions-medium/xv67o6/

评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值