【回溯算法】leetcode 216. 组合总和 III

216. 组合总和 III

题目描述

找出所有相加之和为 n 的 k 个数的组合，且满足下列条件：

• 只使用数字1到9
• 每个数字 最多使用一次

返回 所有可能的有效组合的列表 。该列表不能包含相同的组合两次，组合可以以任何顺序返回。

示例1：

输入： k = 3, n = 7
输出： [[1,2,4]]
解释：
1 + 2 + 4 = 7
没有其他符合的组合了。

示例2：

输入： k = 3, n = 9
输出： [[1,2,6], [1,3,5], [2,3,4]]
解释：
1 + 2 + 6 = 9
1 + 3 + 5 = 9
2 + 3 + 4 = 9
没有其他符合的组合了。

示例3：

输入： k = 4, n = 1
输出： []
解释：
不存在有效的组合。
在[1,9]范围内使用4个不同的数字，我们可以得到的最小和是1+2+3+4 = 10，因为10 > 1，没有有效的组合。

提示

• $2<=k<=9$
• $1<=n<=60$

方法：回溯算法

解题思路

使用回溯三部曲解题：

1. 确定递归函数参数
   和 77. 组合 一样，依然需要一维数组 path 来存放符合条件的结果，二维数组 result 来存放结果集。还需要 sum（int）为已经收集的元素的总和，也就是 path 里元素的总和。

   这里我依然定义 sum、path 和 result 为全局变量。
   接下来还需要如下参数：

   • 目标和，也就是题目中的n。

   • k（int）就是题目中要求k个数的集合。

   • startIndex（int）为下一层for循环搜索的起始位置。

   vector<vector<int>> result;
   vector<int> path;
   int sum = 0;
   void dfs(int startIndex, int n, int k)
   

2. 确定终止条件
   如果 path.size() 和 k 相等了，就终止。

   如果此时 path 里收集到的元素和 sum 和 n 相同了，就用 result 收集当前的结果。

   if(path.size() == k) {
   	if(sum == n)
       	result.push_back(path);
       return;
   }
   

3. 单层搜索过程
   for 循环固定 i <= 9，处理过程就是 path 收集每次选取的元素，相当于树型结构里的边，sum 来统计 path 里元素的总和。处理过程和回溯过程是一一对应的，处理有加，回溯就要有减。

代码

class Solution {
public:
    vector<vector<int>> result;
    vector<int> path;
    int sum = 0;
    void dfs(int startIndex, int n, int k) {
        if(sum > n) return;
        if(path.size() == k) {
            if(sum == n)
                result.push_back(path);
            return;
        }
        for(int i = startIndex; i <= 9; i++) {
            sum += i;
            path.push_back(i);
            dfs(i + 1, n, k);
            path.pop_back();
            sum -= i;
        }
    }
    vector<vector<int>> combinationSum3(int k, int n) {
        dfs(1, n, k);
        return result;
    }
};

复杂度分析

• 时间复杂度：$O((\genfrac{}{}{0ex}{}{M}{k})\times k)$。其中 M 为集合的大小，本题中 M 固定为 9。一共有 $\left(\genfrac{}{}{0ex}{}{M}{k}\right)$ 个组合，每次判断需要的时间代价是 O(k)。
• 空间复杂度：$O(M)$。temp 数组的空间代价是 O(k)，递归栈空间的代价是 O(M)，故空间复杂度为 O(M + k) = O(M).