leetcode 78. 子集

题目描述:给你一个整数数组 nums ,数组中的元素 互不相同 。返回该数组所有可能的子集(幂集)。
解集 不能 包含重复的子集。你可以按 任意顺序 返回解集。

示例:输入:nums = [1,2,3]
输出:[[],[1],[2],[1,2],[3],[1,3],[2,3],[1,2,3]]

题解1(回溯法):难点在于如何能够遍历出所有结果!答案是由递归的终止条件决定! 对于数组就采用以长度为条件的方式去递归。用回溯法解决。用不同的长度作为返回条件。满足长度的就作为答案返回。

leetcode 78. 子集_第1张图片

import copy
class Solution:
    def subsets(self, nums: List[int]) -> List[List[int]]:
        result = []
        result.append([])
        for i in range(1,len(nums)+1):
            self.backtracking(nums,result,i,0,[])
        return result
    def backtracking(self,nums,result,length,index,subset):
        if len(subset) == length:
            tmp = copy.copy(subset)
            result.append(tmp)#这里用subset[:]也行,也可以避免引用拷贝
            return
        for i in range(index,len(nums)):
            subset.append(nums[i])
            self.backtracking(nums,result,length,i+1,subset)
            subset.pop()

题解2(DFS):不撞南墙不回头
leetcode 78. 子集_第2张图片

import copy
class Solution:
    def subsets(self, nums: List[int]) -> List[List[int]]:
        results = []
        self.dfs(nums,results,0,[])
        return results
    
    def dfs(self,nums,results,index,subset):
        results.append(copy.copy(subset))
        if index == len(nums):
            return
        for i in range(index,len(nums)):
            subset.append(nums[i])
            self.dfs(nums,results,i+1,subset)
            subset.pop()

你可能感兴趣的:(刷题,leetcode,算法,职场和发展)