【问题描述】[中等]
【解答思路】
回溯 剪树枝 当和超过n 或 个数超过k
1. 正向求和 优化前
class Solution {
public List<List<Integer>> ans = new LinkedList();
public List<List<Integer>> combinationSum3(int k, int n) {
dfs(0,k,0,n,1,new LinkedList<Integer>());
return ans;
}
private void dfs(int depth,int k,int sum,int n,int num,List<Integer> list){
if(depth>k){
return;
}
if(sum>n){
return;
}
if(depth ==k && sum==n){
ans.add(new LinkedList<Integer>(list));
return;
}
for(int i = num;i<=9;i++){
sum+=i;
list.add(i);
dfs(depth+1,k,sum,n,i+1,list);
list.remove(list.size()-1);
sum-=i;
}
}
}
2. 逆向减等0 优化后
public List<List<Integer>> combinationSum3(int k, int n) {
List<List<Integer>> res = new ArrayList<>();
dfs(res, new ArrayList<>(), k, 1, n);
return res;
}
private void dfs(List<List<Integer>> res, List<Integer> list, int k, int start, int n) {
//终止条件,如果满足这个条件,再往下找也没什么意义了
if (list.size() == k || n <= 0) {
//如果找到一组合适的就把他加入到集合list中
if (list.size() == k && n == 0)
res.add(new ArrayList<>(list));
return;
}
for (int i = start; i <= 9; i++) {
//选择当前值
list.add(i);
//递归
dfs(res, list, k, i + 1, n - i);
//撤销选择
list.remove(list.size() - 1);
}
}
【总结】
1. 逆向可以减少 传参数量
2.在函数中调用dfs时,直接初始化变量,不用另外创建新的变量
3.回溯算法相关题目
[Leedcode][JAVA][第46题][全排列][回溯算法]
[Leetcode][第81题][JAVA][N皇后问题][回溯算法]
[Leetcode][第60题][JAVA][第k个排列][回溯][DFS][剪枝]
转载