算法-回溯-组合总和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 时间复杂度