leetcode216. 组合总和 III/dfs

题目:leetcode216. 组合总和 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]]

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

基本思想:dfs

class Solution {
    
    
public:
    vector<vector<int>> res;
    
    vector<vector<int>> combinationSum3(int k, int n) {
    
    
        dfs(n, k, 1, 0, vector<int>());
        return res;
    }
    
    void dfs(int n, int k, int pos, int cur_sum, vector<int> cur){
    
    
        if(cur.size() == k){
    
    
            if(cur_sum == n){
    
    
                res.push_back(cur);
            }
            return;
        }
        for(int i = pos; i <= 9; ++i){
    
    
            cur.push_back(i);
            dfs(n, k, i + 1, cur_sum + i, cur);
            cur.pop_back();
        }
    }
};

猜你喜欢

转载自blog.csdn.net/qq_31672701/article/details/108541884