py返回某集合的所有子集

Posted 2022-12-10 21座的胖子

题目

思路

一开始的思路像是数学的排列组合，通过切片生成子集，但是考虑了一下还是比较复杂，在集合比较长的时候很难去设计算法。

正确的思路：
比如输入的nums是[1,2,3]

集合[]的子集是：		
[]		数量为1
集合[1]的子集是：
[],[1]		数量为2
集合[1,2]的子集是：
[],[1],[2],[1,2]		数量为4
集合[1,2,3]的子集是：
[],[1],[2],[1,2],[3],[1,3],[2,3],[1,2,3]		数量为8

从上往下看，集合每增加一个元素，新的子集的前一半子集和原集合的子集完全相同，增加的后一半子集是在前一半子集的基础上每个子集增加了新的元素得到。

代码

class Solution:
    def subsets(self, nums: List[int]) -> List[List[int]]:
        a = [[]]	##任何集合的子集都必有空集
        for i in range(len(nums)):	##遍历nums
            s = 0		##设置循环停止条件
            for j in a:		##遍历每次已经存在的a
                k = j[:]	##复制J，必须使用切片，如果直接使用K=J在改变K的时候会改变J
                k.append(nums[i])
                a.append(k)
                s += 1
                if s == 2**i:
                    break
        return a