题目链接:
给定一个可能包含重复元素的整数数组 nums,返回该数组所有可能的子集(幂集)。
说明:解集不能包含重复的子集。
示例:
- 输入: [1,2,2]
- 输出: [ [2], [1], [1,2,2], [2,2], [1,2], [] ]
这道题涉及回溯,回溯的话把问题抽象成树结构会更方便理解。树的图如下:
抽象成树结构后我们可知遍历这个树的时候,把所有节点都记录下来,就是要求的子集集合。像之前求得组合类的问题我们要记录的就是树的叶子节点的数据,而这里的子集是记录所有节点的数据,这就是求组合于子集在抽象成树结构后的区别。
涉及回溯可以利用递归四部曲:
递归作用:从nums中的索引为startIndex的元素开始取往后子集
递归参数:nums意义同题意,startIndex是这一层取数的起始点
终止条件:startIndex==nums.length
单层逻辑:先收集目前记录到的子集,再取这一层对应的数扩充子集,再往下一层取
这一题的难点在于去重,我用了used数组进行去重,具体去重操作代码中有体现
回溯算法记得操作完这一层后要初始化状态量进行回溯
代码如下:
class Solution {
List<List<Integer>> result = null;//收集最终所有的结果
List<Integer> node = null;//记录当前的子集
boolean[] used = null;//去重用的数组
void subsets1(int[] nums, int startIndex){
//递归作用:从nums中的索引为startIndex的元素开始取往后子集
//递归参数:nums意义同题意,startIndex是这一层取数的起始点
//终止条件:startIndex==nums.length
//单层逻辑:先收集目前记录到的子集,再取这一层对应的数扩充子集,再往下一层取
result.add(new ArrayList(node));//收集目前记录到的子集,写在这可以在第一层收集到[]
//if(startIndex == nums.length) return;//其实这句可以不要,因为我们要遍历的是整棵树的节点,下面for里的判断条件就可以当终止条件了
for(int i = startIndex; i < nums.length; i++){
if(i > 0 && nums[i] == nums[i - 1] && used[i - 1] == false) continue;//对树层去重,实现集合与集合之间不同的效果。如果是used[i - 1] == false就是对树枝去重,效果相当于每个组合中不能出现相同的元素。
node.add(nums[i]);//取这一层对应的数扩充子集
used[i] = true;
subsets1(nums, i + 1);//往下一层取数
node.remove(node.size() - 1);//回溯
used[i] = false;
}
}
public List<List<Integer>> subsetsWithDup(int[] nums) {
result = new ArrayList<>();
node = new ArrayList<>();
used = new boolean[nums.length];
Arrays.sort(nums);//排序方便去重
subsets1(nums, 0);
return result;
}
}