给定一个可能包含重复元素的整数数组nums
,返回该数组所有可能的子集(幂集)。
说明:解集不能包含重复的子集。
输入: [1,2,2]
输出:
[
[2],
[1],
[1,2,2],
[2,2],
[1,2],
[]
]
/**
* @param {number[]} nums
* @return {number[][]}
*/
var subsetsWithDup = function(nums) {
var target = [[]];
var n = nums.length;
if(!n) return target;
nums.sort((a, b) => a-b);
var dfs = (cur, tmp, deep, limit) => {
if (tmp.length + (n - cur + 1) < limit) return void 0;
if(limit === deep) {
target.push(tmp);
return void 0;
}
for(let i=cur;i<n; ++i){
if(i>cur && nums[i-1] === nums[i]) continue;
dfs(i+1, [...tmp, nums[i]], deep+1, limit);
}
}
nums.forEach((v,i) => dfs(0, [], 0, i+1));
return target;
};
在本质上是一个组合问题,以一个长度为4
的数组[1, 2, 3, 4]
组合2
个值为例,每两个组合一个数组可取1
组合其数组中之后的值,2
与其数组中之后值,3
与其数组中之后的值,4
与其数组中之后值,即[1, 2]
、[1, 3]
、[1, 4]
、[2, 3]
、[2, 4]
、[3, 4]
,按照这个思路就需要取出给定数组的1 ~ length
长度的组合,这是在给定的数组中没有重复值的情况下,题目中要求会有重复的值,所以在加入的时候我们就需要对其进行操作,首先我们对其进行排序,这样重复的值就会在一起,之后判定对于给定目标长度的数组重复的值只加入一个即可。首先定义目标数组,空数组是所有的数组的子集,所以将空数组置入,之后取得传入的数组的长度n
,如果长度为0
则直接返回目标数组,之后对其进行排序,之后定义深度递归遍历,首先进行剪枝,如果当前tmp
数组的大小为s
,未确定状态的区间[cur,n]
的长度为t
,如果s + t < limit
,那么即使t
个都被选中,也不可能构造出一个长度为limit
的序列,故这种情况就没有必要继续向下递归,之后判断递归深度如果与limit
相等则直接将tmp
数组置入目标数组并返回,之后定义一个循环,在这里我们要处理数字重复的情况,先前已经对其进行排序,所以每次递归后的循环对于数组中重复的值,我们只将第一个置入数组,其他的都忽略,从cur
开始到n
进行递归取值,将tmp
数组与cur
构建一个新数组传递到下一个递归中,之后定义一个循环取得要取得的子集的数组长度,启动递归初始化cur
为0
,深度deep
为0
,tmp
为一个空数组,limit
为i+1
,递归完成后返回目标数组即可。
https://github.com/WindrunnerMax/EveryDay
https://leetcode-cn.com/problems/subsets-ii/