leetcode-中等题-216. 组合总和 III(回溯2)

第一次感觉我的思路比参考答案要清晰一点诶。
清晰在递归函数的终止条件那儿,这样就不需要传入这么多的参数了。
清晰的原因是我没有用到减枝操作,终止条件里,需要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();
}
  • 0
    点赞
  • 0
    收藏
    觉得还不错? 一键收藏
  • 0
    评论

“相关推荐”对你有帮助么?

  • 非常没帮助
  • 没帮助
  • 一般
  • 有帮助
  • 非常有帮助
提交
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值