py返回某集合的所有子集
Posted 21座的胖子
tags:
篇首语:本文由小常识网(cha138.com)小编为大家整理,主要介绍了py返回某集合的所有子集相关的知识,希望对你有一定的参考价值。
题目
思路
一开始的思路像是数学的排列组合,通过切片生成子集,但是考虑了一下还是比较复杂,在集合比较长的时候很难去设计算法。
正确的思路:
比如输入的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
以上是关于py返回某集合的所有子集的主要内容,如果未能解决你的问题,请参考以下文章