Leetcode 216. 组合总和 III

Leetcode 216. 组合总和 III

题目

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

只使用数字1到9
每个数字 最多使用一次
返回 所有可能的有效组合的列表 。该列表不能包含相同的组合两次，组合可以以任何顺序返回。

思路

• 确定递归函数的参数：需要一个一维数组path存放每一个结果，二维数组result存放结果集，这两个定义为全局变量，还需要参数目标和targetSum,k(k个数集合)，sum(已经收集元素的总和)，startIndex(表示为下一层for循环搜索的起始位置)
• 确定终止条件：当path.size() = k时递归终止，并且如果path得元素之和sum等于targetSum的话，使用result收集当前的结果，然后return,进行回溯，准备下一层递归backTracking()
• 单层搜索过程：for循环开始遍历九个数，path收集每一次选取的元素，sum计算path元素的总和

代码

class Solution {
public:
    vector<int> path;// 存放符合条件的结果 存放树的路径
    vector<vector<int>> result;// 存放结果集

    void backTracking(int targetSum,int k,int sum,int startIndex)
    {
        if(path.size() == k)
        {
            // 当path中的数字个数 达到k 检查一下数字之和是否达到目标和
            if(sum == targetSum)
            {
                // 存放path结果
                result.push_back(path);
                return;// 本层递归结束
            }
        }

        for(int i = startIndex; i <= 9; i++)
        {
            // 循环本层集合 进行递归
            sum += i;
            path.push_back(i);
            backTracking(targetSum,k,sum,i+1);// 递归下一个层

            // 当一层递归结束之后 return 代码回到这里：递归语句的下一条  然后进行回溯

            sum -= i;// 回溯
            path.pop_back();// 回溯
        }
    }
    vector<vector<int>> combinationSum3(int k, int n) {
        result.clear();
        path.clear();
        backTracking(n,k,0,1);
        return result;
    }
};