78. 子集 Python

最新推荐文章于 2025-01-22 10:29:55 发布

原创最新推荐文章于 2025-01-22 10:29:55 发布 · 390 阅读

0 ·

CC 4.0 BY-SA版权

文章标签：

#python #算法 #leetcode

LeetCode中等难度题集专栏收录该内容

104 篇文章

订阅专栏

文章介绍了如何使用Python的itertools库中的combinations方法来生成一个整数数组的所有不重复子集，提供了代码示例和解题思路。给定一个互不相同的整数数组nums，程序依次生成从空集到全集的所有子集，并存储在结果列表中。

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

一、题目描述

给你一个整数数组 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]]:
        length =len(nums)
        result = []
        path = []
        for i in range(length+1):
            print(f"count: {i}")
            path = list(itertools.combinations(nums,i))
            for i in range(len(path)):
                path[i] = list(path[i])
            result.extend(path)
            print(result)

        print("result:",result)
        return result

三、解题思路

本题解题思路为依次求数组nums从长度为0的全组合列表([[]])到长度为length的全组合列表(例如：nums=[1,2,3]，length=3时的全组合序列只有[1,2,3])，并全部添加进入result列表中。
求全组合的方法是采用itertools库中的combinations方法，细节处理可以查看链接: 77.组合 Python
需要注意的是，本题由于限制了nums数组的长度在10以内，所以采用解采用combinations方法可以快速求得组合。