题目来源:https://leetcode.cn/problems/subsets-ii/
大致题意:
给一个数组,求出数组的所有子集,子集中的元素顺序忽略不计
思路
题目要求子集的元素顺序忽略不计,那么顺序不同元素相同即算同一个子集
于是可以先对数组排序,这样在递归的时候就可以防止重复元素子集的出现
具体递归的规则为:
- 递归时使用一个 list 存下当前子集元素,一个标志位标记上一个元素是否在子集中,以及当前的遍历索引
- 进入递归函数首先判断当前索引是否等于数组长度,若等于直接将当前子集加入答案集合
- 接下来可以跳过当前元素,直接递归下一个位置
- 判断上一个元素是否在子集中以及当前位置是否与上一个位置元素值相同,若上一个元素不在子集且元素值相同,那么不用放入当前元素(因为在上一个位置会递归放入当前元素的子集);否则,将当前位置元素加入子集,递归下一个位置
代码:
class Solution {
List<List<Integer>> ans; // 答案集合
int[] nums;
int n;
public List<List<Integer>> subsetsWithDup(int[] nums) {
ans = new ArrayList<>();
this.nums = nums;
n = nums.length;
Arrays.sort(nums); // 先排序
dfs_noSet(0, false, new ArrayList<>());
return ans;
}
public void dfs_noSet(int idx, boolean choosePre, List<Integer> list) {
// 递归结束条件
if (idx == n) {
ans.add(new ArrayList<>(list));
return;
}
// 跳过当前元素
dfs_noSet(idx + 1, false, list);
// 如果上一个元素未放入且当前元素于上一个元素值相同,不放入本位置元素
if (idx != 0 && !choosePre && nums[idx] == nums[idx - 1]) {
return;
}
// 加入当前位置元素
list.add(nums[idx]);
dfs_noSet(idx + 1, true, list);
// 回溯
list.remove(list.size() - 1);
}
}