python 刷leetcode 题目（34）

最新推荐文章于 2024-07-31 09:52:57 发布

原创最新推荐文章于 2024-07-31 09:52:57 发布 · 635 阅读

0 ·

CC 4.0 BY-SA版权

文章标签：

#全排列 #子集 #leetcode

python 专栏收录该内容

64 篇文章

订阅专栏

本文介绍了解决子集生成及全排列问题的两种高效算法。对于子集生成，提供了迭代与递归两种方法；而对于全排列问题，则采用递归方式解决。这两种经典算法能够帮助理解组合数学中重要的概念。

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >

78. 子集

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

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

示例:

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

代码;

class Solution3(object):
    def subsets(self, nums):
        """
        :type nums: List[int]
        :rtype: List[List[int]]
        """
        # res = [[]]
        # for num in nums:
        #     for temp in res[:]:
        #         x = temp[:]
        #         x.append(num)
        #         res.append(x)
        # return res

    #### the second method
    #### 类似于二叉树的前序遍历， 用DFS可以实现
        self.results = []
        self.search(sorted(nums), [], 0)
        return self.results

    def search(self, nums, S, index):
        if index == len(nums):
            self.results.append(S)
            return

        self.search(nums, S + [nums[index]], index + 1)
        self.search(nums, S, index + 1)

s = Solution3()
print(s.subsets([1,2,3]))

全排列

给定一个没有重复数字的序列，返回其所有可能的全排列。

示例:

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

代码如下：

class Solution(object):
    def permute(self, nums):
        """
        :type nums: List[int]
        :rtype: List[List[int]]
        """
        if len(nums) <= 1:
            return [nums]
        res = []
        for i, num in enumerate(nums):
            n = nums[: i] + nums[i+1 :]
            for x in self.permute(n):
                res.append([num] + x)
        return res