Leetcode-78-子集

题目说明

给你一个整数数组 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 中的所有元素 互不相同

来源：力扣（LeetCode） 链接：https://leetcode.cn/problems/subsets

题解

回溯法

def backtracking(nums, result, length, index, subset):
    if length == len(subset):
        result.append(subset.copy())
        return
    for i in range(index, len(nums)):
        subset.append(nums[i])
        backtracking(nums, result, length, i+1, subset)
        subset.pop()


def fun(nums):
    result = []
    result.append([])
    for i in range(1, len(nums) + 1):
        backtracking(nums, result, i, 0, [])
    return result

print(fun([1,2,3]))

执行结果：

[[], [1], [2], [3], [1, 2], [1, 3], [2, 3], [1, 2, 3]]

DFS

def dfs(nums, result, index, subset):
    result.append(subset.copy())
    for i in range(index, len(nums)):
        subset.append(nums[i])
        dfs(nums, result, i+1, subset)
        subset.pop()

def fun(nums):
    result = []
    dfs(nums, result, 0, [])
    return result

print(fun([1,2,3]))

执行结果：

[[], [1], [1, 2], [1, 2, 3], [1, 3], [2], [2, 3], [3]]