回溯算法---子集系列

力扣 78. 子集

原理

和力扣 77.组合 问题原理相似，不一样的在于收获结果的位置不一样，在子集问题中，每个节点都可以收获结果，而不像组合、分割、全排列问题只会在叶子节点收获结果；另外就是最小子问题不一样

JS实现

var subsets = function (nums) {
  let res = []
  let path = []
  const backtracking = function (startIndex) {
    if (startIndex >= nums.size) return
    for (let i = startIndex; i < nums.length; i++) {
      path.push(nums[i])
      // 所有节点都需要收获结果
      res.push([...path])
      backtracking(i + 1)
      path.pop()
    }
  }
  res.push([])
  backtracking(0)
  return res
}

90. 子集 II

原理

数组中包含重复元素，需要进行树层去重，去重逻辑类似于组合、全排列问题中遇到的，都需要维护一个数组用来记录元素的使用情况，上个栗子，
输入：nums = [1,2,2]
输出：[[],[1],[1,2],[1,2,2],[2],[2,2]]
其转化为树形结构为：

JS实现

var subsetsWithDup = function (nums) {
  let res = []
  let path = []
  let used = new Array(nums.length).fill(false)
  // 排序，为后面去重做准备
  nums.sort((a, b) => a - b)
  const backtracking = function (startIndex) {
    if (startIndex >= nums.length) return
    for (let i = startIndex; i < nums.length; i++) {
      // 树层去重
      if (i > 0 && nums[i] === nums[i - 1] && used[i - 1] === false) continue
      path.push(nums[i])
      // 所有节点都要收获结果
      res.push([...path])
      used[i] = true
      backtracking(i + 1)
      path.pop()
      used[i] = false
    }
  }
  res.push([])
  backtracking(0)
  return res
}