LeetCode78_子集(三种解法)

1. 题目

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

2. 题解

# 78
from typing import List
import copy


class Solution:
    # 方法1: 扩展法
    def subsets_1(self, nums: List[int]) -> List[List[int]]:
        result = []
        result.append([])
        for n in nums:
            temp = []
            for res in result:
                r = copy.deepcopy(res)
                r.append(n)
                temp.append(r)

            for t in temp:
                result.append(t)
        return result

    # 方法2: 回溯法
    def subsets_2(self, nums: List[int]) -> List[List[int]]:
        result = []
        result.append([])
        for i in range(1, len(nums) + 1):
            self.backtracking(nums, result, i, 0, [])   # i 表示数组长度, index 表示当前序号
        return result

    def backtracking(self, nums, result, length, index, subset):
        if len(subset) == length:
            temp = copy.deepcopy(subset)
            result.append(temp)
            return

        for i in range(index, len(nums)):
            subset.append(nums[i])
            self.backtracking(nums, result, length, i + 1, subset)
            subset.pop()

    # 方法3: 分治法
    def subsets_3(self, nums: List[int]) -> List[List[int]]:
        result = []
        self.dfs(nums, result, 0, [])
        return result

    def dfs(self, nums, result, index, subset):
        result.append(copy.deepcopy(subset))
        if index == len(nums):
            return

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

你可能感兴趣的:(LeetCode)