原题链接:https://leetcode-cn.com/problems/subsets-ii/
解题思路:
- 使用DFS生成所有可能的排列情况。
- 每层递归从
current
开始遍历nums
。 - 将遍历到的值存入
subset
,并进入下一层递归,表示当前值存在与子集中的情况。 - 进入下一层递归后,将当前值从
subset
中弹出,继续遍历数组,表示当前值不存在与子集中的情况。 - 如何去重:
- 以输入
[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;
};