【问题描述】[中等]
【解答思路】
回溯 剪树枝 当和超过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][剪枝]
转载