Leetcode 216. 组合总和 III C++

Leetcode 216. 组合总和 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]]

题解

dfs回溯
用一个变量记录选中的数的和，如果大于目标值或大于k个数，则返回；等于，且正好选了k个数，则放入答案中；小于，则接着进行选择。详细过程见代码

代码

 	vector<vector<int>> ans;
    void dfs(int begin,int time,int k,int sum,int n,vector<int>& now){
        if(time == k && sum==n){	//选了k个数，和为n，则把选的数放入答案中
            ans.push_back(now);
        }else if(time < k){
            if(sum >= n)    return;		//和大于目标值，返回；等于目标值，再选一个数肯定会大于目标值，也返回
            for(int i=begin; i<10; i++){
                now.push_back(i);
                dfs(i+1,time+1,k,sum+i,n,now);
                now.pop_back();
            }
        }
    }
    vector<vector<int>> combinationSum3(int k, int n) {
        vector<int> now;
        dfs(1,0,k,0,n,now);
        return ans;
    }

来源：力扣（LeetCode）
链接：https://leetcode-cn.com/problems/combination-sum-iii
著作权归领扣网络所有。商业转载请联系官方授权，非商业转载请注明出处。