关于回溯法的模版请看:https://www.jianshu.com/p/2a9856b96a86
leetcode 78 子集
给你一个整数数组 nums ,数组中的元素 互不相同 。返回该数组所有可能的子集(幂集)。解集 不能 包含重复的子集。你可以按 任意顺序 返回解集
这个题和组合总和的题类似,只不过我们将解集收集的地方不同,套回溯法模版
public List<List<Integer>> subsets(int[] nums) {
List<List<Integer>> res = new ArrayList<>();
Arrays.sort(nums);
subsetsHelp(res, new ArrayList<>(), nums, 0);
return res;
}
private void subsetsHelp(List<List<Integer>> res, ArrayList<Integer> list, int[] nums, int index) {
// 在递归开始的地方将路径进行收集,一边回溯一边收集
res.add(new ArrayList<>(list));
for (int i = index; i < nums.length; i++) {
list.add(nums[I]);
subsetsHelp(res, list, nums, i + 1);
list.remove(list.size()-1);
}
}
Leetcode 90 子集2
给定一个可能包含重复元素的整数数组 nums,返回该数组所有可能的子集(幂集)。
该题的解法和组合总和2基本相同
public List<List<Integer>> subsetsWithDup(int[] nums) {
List<List<Integer>> res = new ArrayList<>();
Arrays.sort(nums);
subsetsWithDupHelp(res, new ArrayList<>(), nums, 0);
return res;
}
private void subsetsWithDupHelp(List<List<Integer>> res, ArrayList<Integer> list, int[] nums, int index) {
res.add(new ArrayList<>(list));
for (int i = index; i < nums.length; i++) {
// 同一层若相同元素被使用过则跳过
if (i > index && nums[i] == nums[i-1]) {
continue;
}
list.add(nums[I]);
subsetsWithDupHelp(res, list, nums, i+1);
list.remove(list.size()-1);
}
}
网友评论