LeetCode 78.子集

题目

给定一组不含重复元素的整数数组 nums，返回该数组所有可能的子集（幂集）。

说明：解集不能包含重复的子集。

示例:

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

思路

这道题可以用回溯法，就是说如果碰到不满足条件的就返回上一级寻找下一个可用答案，下面贴上一个讨论区里的回溯法标准模板来具体分析。

def backtrack(nums, res, temp, start):
    res.append(temp)
    for i in range(start, len(nums)):
        if nums[i] not in temp:
            backtrack(nums, res, temp + [nums[i]], i+1) # 这个递归调用里，res是直接传递列表，所以在递归的过程中res会被永久地改变，
                                                        # 而temp+[nums[i]]的语法传入的其实是一个副本，所以这里面temp没有改变，还是调用函数之前的值

class Solution(object):
    def subsets(self, nums):
        """
        :type nums: List[int]
        :rtype: List[List[int]]
        """
        res = []
        backtrack(nums, res, [], 0)
        return res

输出是这样，按照这种排布可能规律看的清楚一些。做完[1,2,3,4,5]之后，会回退到[1,2,3]状态，之后继续for循环，i变成4(之前i=3的时候做过了，i=3才会有[1,2,3,4,5])，所以会append 5，之后又回退跑到[1,2]，然后i = 3，所以append 4，进入下一层递归，于是有[1,2,4,5]，等等依此类推。

[[], 
 [1], [1, 2], [1, 2, 3], [1, 2, 3, 4], [1, 2, 3, 4, 5], [1, 2, 3, 5], [1, 2, 4], [1, 2, 4, 5], [1, 2, 5], [1, 3], [1, 3, 4], [1, 3, 4, 5], [1, 3, 5], [1, 4], [1, 4, 5], [1, 5], 
 [2], [2, 3], [2, 3, 4], [2, 3, 4, 5], [2, 3, 5], [2, 4], [2, 4, 5], [2, 5], 
 [3], [3, 4], [3, 4, 5], [3, 5], 
 [4], [4, 5], 
 [5]]

循环方法class Solution:

class Solution:
    def subsets(self, nums: List[int]) -> List[List[int]]:
        res = [[]]
        for num in sorted(nums):
            res += [item+[num] for item in res]
        return res

这个方法主要解释一下res += [item+[num] for item in res]这一行，意思是每次给res的末尾添上一个num，第一轮添1，第二轮添2，第三轮添3……
输出如下：

[[], 
[1], 
[2], [1, 2], 
[3], [1, 3], [2, 3], [1, 2, 3], 
[4], [1, 4], [2, 4], [1, 2, 4], [3, 4], [1, 3, 4], [2, 3, 4], [1, 2, 3, 4], 
[5], [1, 5], [2, 5], [1, 2, 5], [3, 5], [1, 3, 5], [2, 3, 5], [1, 2, 3, 5], [4, 5], [1, 4, 5], [2, 4, 5], [1, 2, 4, 5], [3, 4, 5], [1, 3, 4, 5], [2, 3, 4, 5], [1, 2, 3, 4, 5]]