LeetCode 216. 组合总和 III

题目:216. 组合总和 III

类型:回溯算法

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

说明:

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

示例 1:

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

示例 2:

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

解题思路:

这道题的解题思路和 77题相同,只需要加上添加数组时的判断条件和优化条件即可。

代码:

    //0ms 超过100%
    vector> combinationSum3(int k, int n) {
        
        vector> res;
        vector nums;
        recursion(n, k, 1, 10, res, nums);
        return res;
    }
    
    
    void recursion(int n, int k, int left, int right, vector>& res, vector nums){
        
        //保存当前数组的条件为达到目标值且位数为 k位
        if(!k && !n) res.push_back(nums);
        else{
            for(int i = left; i < right; i++){

                //优化条件,当当前的总和已大于目标值时,剩余的数列不需要循环
                if(n-i < 0) return;
                nums.push_back(i);
                recursion(n-i, k-1, i+1, right, res, nums);
                nums.pop_back();
            }   
        }
    }

 

你可能感兴趣的:(算法——回溯算法)