题目一:78. 子集
题目思路
回溯三部曲
-
递归函数参数
全局变量数组path
为子集收集元素,二维数组result
存放子集组合。(也可以放到递归函数参数里)
递归函数参数在上面讲到了,需要startIndex
。
-
递归终止条件
剩余集合为空的时候,就是叶子节点。
那么什么时候剩余集合为空呢?
就是startIndex已经大于数组的长度了,就终止了,因为没有元素可取了,代码如下:
if (startIndex = nums.size()) {return;}
其实可以不需要加终止条件,因为startIndex >= nums.size(),本层for循环本来也结束了。
-
单层搜索逻辑
求取子集问题,不需要任何剪枝!因为子集就是要遍历整棵树。
那么单层递归逻辑代码如下:
for (int i = start; i < nums.length; i++) {
int num = nums[i];
path.push(num);
dfs(i + 1, nums, list, path); // 注意从i+1开始
path.pop();
}
题解代码
public List<List<Integer>> subsets(int[] nums) {
List<List<Integer>> list = new ArrayList<>();
LinkedList<Integer> path = new LinkedList<>();
dfs(0, nums, list, path);
return list;
}
private void dfs(int start, int[] nums, List<List<Integer>> list, LinkedList<Integer> path) {
// 子集是收集树形结构中树的所有节点的结果。
list.add(new ArrayList<>(path));
for (int i = start; i < nums.length; i++) {
int num = nums[i];
path.push(num);
dfs(i + 1, nums, list, path); // 注意从i+1开始
path.pop();
}
}
题目二:90. 子集 II
题目思路
这种需要回溯解决的题目,组合问题,子集问题,切割问题,如果遇到原先给你的集合中有重复元素,而输出的结果中不允许有重复结果的,需要先将原始数组排序,然后使用一个boolean类型的数组,标记每个元素是否被访问,对于重复的元素,如果前一个元素没有被访问,则不允许访问第二个元素
题解代码
List<List<Integer>> result = new ArrayList<>();
LinkedList<Integer> path = new LinkedList<>();
boolean[] used; // 元素是否被访问
public List<List<Integer>> subsetsWithDup(int[] nums) {
if (nums.length == 0) {
result.add(path);
return result;
}
Arrays.sort(nums);
used = new boolean[nums.length];
dfs(nums, 0);
return result;
}
private void dfs(int[] nums, int startIndex) {
result.add(new ArrayList<>(path));
if (startIndex >= nums.length) { // 终止条件
return;
}
for (int i = startIndex; i < nums.length; i++) {
// 去重,递归树层去重,如果nums中有重复,则必须先访问前面的元素才能访问后面的元素
if (i > 0 && nums[i] == nums[i - 1] && !used[i - 1]) {
// 如果出现重复,前面一个还没访问,则跳出循环
continue;
}
int num = nums[i];
path.push(num);
used[i] = true;
dfs(nums, i + 1);
path.pop();
used[i] = false;
}
}