78. 子集 Python


一、题目描述

给你一个整数数组 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方法可以快速求得组合。


版权声明:本文为weixin_44807903原创文章,遵循CC 4.0 BY-SA版权协议,转载请附上原文出处链接和本声明。