一个回溯算法的例子
给定一个候选人编号的集合 candidates 和一个目标数 target ,找出 candidates 中所有可以使数字和为 target 的组合。
candidates 中的每个数字在每个组合中只能使用 一次 。
注意:解集不能包含重复的组合。
示例 1:
输入: candidates = [10,1,2,7,6,1,5], target = 8,
输出:
[
[1,1,6],
[1,2,5],
[1,7],
[2,6]
]
示例 2:
输入: candidates = [2,5,2,1,2], target = 5,
输出:
[
[1,2,2],
[5]
]
提示:
1 <= candidates.length <= 100
1 <= candidates[i] <= 50
1 <= target <= 30
错误的做法
class Solution {
List<List<Integer>> ans =new ArrayList<List<Integer>>();
public List<List<Integer>> combinationSum2(int[] candidates, int target) {
// #构建二叉树
// 结合剪枝技术,深度优先遍历这个树
Arrays.sort(candidates);
dfs(0,new ArrayList<Integer>(),target,candidates);
return ans;
}
public void dfs(int idx,ArrayList<Integer> path,int remin,int[] candidates){
for (int i=idx;i<candidates.length;i++){
int c=candidates[i];
// 去重操作
if (i>idx){
if (candidates[i] == candidates[i - 1]){
continue;
}
}
if (c>remin){
break;
}
if (c==remin){
path.add(c);
ans.add(new ArrayList<Integer>(path));
return;
}
if (c<remin){
path.add(c);
// 这里不能直接操作path,即将path.add(*)后再调用。因为这样就破坏了path在这个函数内的引用
dfs(i+1,path,remin-c,candidates);
};
}
}
}
如果上面的写法就错了,因为我们回溯的过程中关键的参数是path,回溯回来的时候应该跟将dfs压入栈中的参数一致,但是path.add(c );就破坏了这个结构。
所以正确的做法是下面这样:
import java.lang.reflect.Array;
import java.util.ArrayList;
import java.util.Arrays;
import java.util.List;
public class Solution {
List<List<Integer>> ans =new ArrayList<List<Integer>>();
public List<List<Integer>> combinationSum2(int[] candidates, int target) {
// #构建二叉树
// 结合剪枝技术,深度优先遍历这个树
Arrays.sort(candidates);
// for (int i=0;i<candidates.length;i++){
// System.out.println(candidates[i]);
// };
dfs(0,new ArrayList<Integer>(),target,candidates);
return ans;
}
public void dfs(int idx,ArrayList<Integer> path,int remin,int[] candidates){
for (int i=idx;i<candidates.length;i++){
int c=candidates[i];
// 去重操作
if (i>idx){
if (candidates[i] == candidates[i - 1]){
continue;
}
}
if (c>remin){
break;
}
if (c==remin){
path.add(c);
ans.add(new ArrayList<Integer>(path));
return;
}
if (c<remin){
ArrayList<Integer> thisArray=new ArrayList<Integer>(path);
thisArray.add(c);
// 这里不能直接操作path,即将path.add(*)后再调用。因为这样就破坏了path在这个函数内的引用
dfs(i+1,thisArray,remin-c,candidates);
};
}
}
public static void main(String args[]){
System.out.println("ddd");
Solution s=new Solution();
int[] cand={10,1,2,7,6,1,5};
System.out.println(s.combinationSum2(cand,8));
}
}
在leecode上直接击败了百分之99.76