回溯法, 数组的子集

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

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

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

链接：https://leetcode-cn.com/problems/subsets

---

在快手的面试中遇到了这个题, 知道需要用回溯法枚举出所有的场景, 但是在写的过程中, 还是有点卡壳.

先把最开始的思路写上去, 用一个数组存放一次枚举出的结果, 另一个数组存放剩余未确定的元素, 当剩余未确定的元素为0时, 说明这一次的枚举完成了 ,在一个result数组中保存本次结果.

在每次枚举的过程中, 每个元素有2种情况, 加入到数组中, 或者不加入. 

class Solution {
    
    var result = [[Int]]()
    
    /// 递归调用, 生成每次的枚举的结果
    ///   - currentResult: 已经确定的元素
    ///   - leftArray: 剩下的未确定是否加入的元素
    func showAllResult(_ currentResult: inout [Int], _ leftArray: [Int]) {
        // 剩下的元素==0, 说明此次结果已经完成了,加入结果到set中
        if leftArray.count==0 {
            //print(currentResult)
            result.append(currentResult)
            return
        }
        
        // 对剩余元素, 每个元素有加入或者不加入2种情况
        
        // 处理item不加入结果的情况
        var temp = Array(leftArray[1..<leftArray.count])
        showAllResult(&currentResult, temp)
        
        // 处理item加入到结果的情况
        currentResult.append(leftArray.first!)
        showAllResult(&currentResult, temp)
        // 回溯法的精髓, 回退上次结果
        currentResult.removeLast()
        
    }
    
    func subsets(_ nums: [Int]) -> [[Int]] {
        
        var oneResult = [Int]()
        showAllResult(&oneResult, nums)
        
        return result
        
    }
}

---

官方题解上还有一种使用遍历的方法, 效率也不错. 

一个集合元素为n个, 那么子集就是2^n个, 从0~2^n -1中的数字用2进制表示正好是所有的情况，

假设有3个数字[5,2,9],  i从0~7换成2进制表示就是0x000,0x001,0x010,0x011,0x100,0x101,0x110,0x111, 用000表示子集[],001可以表示子集[9], 011表示[2,9],111表示[5,2,9],每个i就对应一种子集, 当遍历完成时, 结果也就出来了 

class Solution {
    
    func subsets(_ nums: [Int]) -> [[Int]] {
        
        // 存放所有结果
        var result = [[Int]]()
        
        // 1循环右移n位 就是2^n,也就是结果的个数
        let times = 1<<nums.count
        // i的取值是从[0,2^n), 假设有3个数字,i换成2进制表示就是0x000,0x001,0x010,0x011,0x100,0x101,0x110,0x111,
        // 用000表示子集[],001可以表示子集[3], 011表示[2,3],111表示[1,2,3],每个i就对应一种子集
        for i in 0..<times {
            
            // 存放单次结果
            var oneResult = [Int]()

            for (j,item) in nums.enumerated() {
                // 判断i的第j位是不是0, 如果是0, 不加入数组; 如果不是0, 加入到结果中
                if (i & (1<<j)) != 0  {
                    oneResult.append(item)
                }
                
            }
            result.append(oneResult)
            
        }
        return result
        
    }
}