算法-回溯-组合总和3

算法-回溯-组合总和3

1 题目概述

1.1 题目出处

https://leetcode-cn.com/problems/combination-sum-iii/submissions/

1.2 题目描述

找出所有相加之和为 n 的 k 个数的组合。组合中只允许含有 1 - 9 的正整数,并且每种组合中不存在重复的数字。

说明:

所有数字都是正整数。
解集不能包含重复的组合。
示例 1:

输入: k = 3, n = 7
输出: [[1,2,4]]
示例 2:

输入: k = 3, n = 9
输出: [[1,2,6], [1,3,5], [2,3,4]]

2 回溯法

2.1 思路

这段转自组合总和-解题思路

回溯法的解体框架是什么呢,解决一个回溯问题,实际上就是一个决策树的遍历过程。一般来说,我们需要解决三个问题:

  • 路径:也就是已经做出的选择。
  • 选择列表:也就是你当前可以做的选择。
  • 结束条件:也就是到达决策树底层,无法再做选择的条件。

其中最关键的点就是:在递归之前做选择,在递归之后撤销选择。

LinkedList result = new LinkedList();
public void backtrack(路径,选择列表){
    if(满足结束条件){
        result.add(结果);
    }
    for(选择:选择列表){
        做出选择;
        backtrack(路径,选择列表);
        撤销选择;
    }
}

所以我们这里只需要按套路来就行了,不过需要注意一些地方:

  • 要将结束条件写在for循环里,才能剪枝,否则会进行不必要递归
  • 遍历路径时发现求得的和已经和目标值n相等时,还需要确定是否满足k个数。

2.2 代码

class Solution {
    List<List<Integer>> resultList = new LinkedList<>();
    public List<List<Integer>> combinationSum3(int k, int n) {
        // 最多就是1到9数字相加为45,超过就不合法
        if(n > 45){
            return resultList;
        }
        backtrack(k, n, 1, new LinkedList<>(), 0);
        return resultList;
    }
    private void backtrack(int k, int n, int start, List<Integer> tmpList, int tmpSum){
        if(k == 0){
            // 结束条件1
            return;
        }
        LinkedList<Integer> newList = new LinkedList<>(tmpList);
        for(int i = start; i < 10; i++){
            // 选择当前数字
            int tmpSum2 = tmpSum + i;
            if(tmpSum2 > n){
                // 结束条件2,剪枝
                break;
            }

            newList.add(i);
            if(tmpSum2 == n ){
                // 结束条件3,剪枝
                if(k == 1){
                    // 只有这个时候才满足k个数之和为n
                    resultList.add(newList);
                }
                break;
            }
            
            backtrack(k-1, n, i + 1, newList, tmpSum2);

            // 不选当前数字
            newList.removeLast();
        }
    }
}

2.3 时间复杂度

在这里插入图片描述

  • 0
    点赞
  • 0
    收藏
    觉得还不错? 一键收藏
  • 0
    评论
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值