第一次感觉我的思路比参考答案要清晰一点诶。
清晰在递归函数的终止条件那儿,这样就不需要传入这么多的参数了。
清晰的原因是我没有用到减枝操作,终止条件里,需要size==k才开始返回,其实可以加入一个提前返回的条件,那就是:
每添加一个值如path,就用sum记录一下总和,如果总和超过了n,那就可以直接返回了,不需要继续加入值到path。
//按照回溯算法处理本题的话,我现在只能想到利用递归函数来处理
// 函数参数:(k, n, 开始遍历位置startIndex)
// 返回:result
// 终止条件:if(path.size == k) return;
// 处理逻辑:
// for(int i = startIndex; i < n - (k - path.size()) + 1; i++){
// path.add(i);
// traverse(k, n, i+1);
// path.removeFirst();
// }
class Solution {
LinkedList<Integer> path = new LinkedList<>();
List<List<Integer>> result = new ArrayList<>();
public List<List<Integer>> combinationSum3(int k, int n) {
traverse(k, n, 1);
return result;
}
public void traverse(int k, int n, int startIndex){
if(path.size() == k){
int sum = 0;
for(Integer i : path){
sum += i;
}
if(sum == n)
result.add(new LinkedList(path));
return;
}
for(int i = startIndex; i <= 9 - (k - path.size()) + 1; i++){
// System.out.println("i =" + i);
path.add(i);
traverse(k, n, i+1);
// System.out.println(path);
path.removeLast();
}
}
}
减枝操作:
if (sum > targetSum) { // 剪枝操作
return; // 如果path.size() == k 但sum != targetSum 直接返回
}
这里的targetSum其实就是n
所以递归函数要多加入一个sum来做判断,当然sum这个参数需要回溯:
for (int i = startIndex; i <= 9 - (k - path.size()) + 1; i++) {
path.add(i);
sum += i;
backTracking(targetSum, k, i + 1, sum);
//回溯
path.removeLast();
//回溯
sum -= i;
}
我觉得这样简化一下就能回溯:因为不改变sum的值:
for (int i = startIndex; i <= 9 - (k - path.size()) + 1; i++) {
path.add(i);
backTracking(targetSum, k, i + 1, sum + i);
//回溯
path.removeLast();
}