Leetcode刷题Python78. 子集

Posted 2022-08-07 Better Bench

1 题目

给你一个整数数组 nums ，数组中的元素 互不相同 。返回该数组所有可能的子集（幂集）。

解集 不能 包含重复的子集。你可以按 任意顺序 返回解集。

示例 1：

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

示例 2：

输入：nums = [0]
输出：[[],[0]]

2 解析

用二进制中的1来表示当前index的元素是否在子集中。例如，n = 3，a=5,2,9 时对应的二进制

0-1二进制序列正好对应子集的索引。1表示取当前索引的值，0表示不取。用与运算来计算，二进制中当前位置是否是1。

3 Python实现

class Solution:
    def subsets(self, nums: List[int]) -> List[List[int]]:
        ans = []
        n = len(nums)
        for mask in range(1<<n):
            t = []
            for i in range(n):
            	# 与运算，来判断二进制序列中，当前位置是否是1
                if (mask & (1<<i))!=0:
                    t.append(nums[i])
            ans.append(t)
        return ans