chencl1986 / Blog

Welcome to lee's blog.

Geek Repo:Geek Repo

Github PK Tool:Github PK Tool

LeetCode题解:90. 子集 II,递归+for循环+回溯,JavaScript,详细注释

chencl1986 opened this issue · comments

原题链接:https://leetcode-cn.com/problems/subsets-ii/

解题思路:

  1. 使用DFS生成所有可能的排列情况。
  2. 每层递归从current开始遍历nums
  3. 将遍历到的值存入subset,并进入下一层递归,表示当前值存在与子集中的情况。
  4. 进入下一层递归后,将当前值从subset中弹出,继续遍历数组,表示当前值不存在与子集中的情况。
  5. 如何去重:
    • 以输入[1,2,2]为例,其在不进行去重的情况下,输出的结果为[[],[1],[1,2],[1,2,2],[1,2],[2],[2,2],[2]]
    • 可以看到重复的子集有[1,2][2],其产生的原因是nums中有2个“2”。
    • 可以看出如果nums中有多个相同值,每次遍历都会产生一组相同的子集。
    • 因此去重的思路是,在遍历nums生成子集时,只要判断相同值已被遍历过,就跳过本次遍历。
    • 在递归时,for循环的起始值,即i = current会被优先执行,因此需要使用i > current的判断,保证i = current的情况一定会执行。
    • 之后任何一次遍历中,如果遇到nums[i] === nums[i - 1],即表示nums[i]的子集已被遍历过,则可以跳过。
    • i = current即为每次递归下探的情况,在其之前的nums中元素都被选中了,因此子集[1,2,2]是不会被漏掉的。
    • i = current的情况是不会被跳过,因此子集[1,2][2]都会存在。
    • [1,2][2]的重复情况都出现在之前的某个元素未被选中之后的遍历,即for循环中位于current之后的遍历,因此需要加上nums[i] === nums[i - 1]判断去重。
    • 在开始递归前,需要将nums进行排序,以保证nums[i] === nums[i - 1]是有效的。
function dfs(nums, subset, result, current) {
  // 每次递归的参数subset,都是在上一层生成的子集,直接将其存入结果
  result.push(subset.slice());

  // 从current开始遍历,current之前的元素都已被遍历过
  // 并且已经被按照是否存储两种情况存入子集
  // i < nums.length充当了递归终止条件
  for (let i = current; i < nums.length; i++) {
    if (i > current && nums[i] === nums[i - 1]) {
      continue;
    }

    // 下一层递归的current
    const newCurrent = i + 1;

    // 将当前值存入子集,并且进入下一层递归
    // 下一层递归处理的是当前值存在与子集中的情况
    subset.push(nums[i]);

    // 进入下一层递归
    dfs(nums, subset, result, newCurrent);

    // 将当前值弹出,继续当前的for循环遍历
    // 当前层的继续遍历,处理了当前值不存在与子集中的情况
    subset.pop();
  }
}
/**
 * @param {number[]} nums
 * @return {number[][]}
 */
var subsetsWithDup = function (nums) {
  let result = []; // 存储结果
  let subset = []; // 存储子集
  // 将nums排序,保证递归中nums[i] === nums[i - 1]的判断有效
  nums.sort((a, b) => a - b);
  // 使用DFS+回溯生成所有子集
  dfs(nums, subset, result, 0);
  return result;
};