LeetCode 216. 组合总和 III（Combination Sum III）

题目描述

找出所有相加之和为 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]]

解题思路

回溯算法，从第一个数开始依次添加数字并比较当前数字总和，若相等就添加到结果集合中。

代码

 class Solution {

 public:

     vector<vector<int>> combinationSum3(int k, int n) {

         vector<vector<int>> res;

         if(k <=  || n <= ) return res;

         vector<int> nums, temp;

         for(int i = ; i < ; i++)

             nums.push_back(i + );

         combine(nums, , , k, n, temp, res);

         return res;

     }

     void combine(vector<int> nums, int idx, int sum, int k, int n, vector<int> &temp, vector<vector<int>> &res){

         if(k ==  && sum == n)

             res.push_back(temp);

         else if(sum < n){

             for(int i = idx; i < ; i++)

                 if(sum + nums[i] <= n){

                     temp.push_back(nums[i]);

                     combine(nums, i + , sum + nums[i], k - , n, temp, res);

                     temp.pop_back();

                 }

         }

     }

 };

LeetCode 216. 组合总和 III（Combination Sum III）的相关教程结束。