leetcode 216.组合总和|||（dfs经典题）

找出所有相加之和为 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>> res;
    vector<int> path;

    void dfs(int k,int n,int start,int sum)
    {
    
    
        if(path.size()==k) 
        {
    
    
            if(sum==n)
            res.push_back(path);
            return;
        }
        for(int i=start;i<=9;i++)
        {
    
    
            
            path.push_back(i);
            sum+=i;
            dfs(k,n,i+1,sum);
            sum-=i;
            path.pop_back();
        }
    }

    vector<vector<int>> combinationSum3(int k, int n) {
    
    
        dfs(k,n,1,0);
        return res;
    }
};

剪枝优化做法：

class Solution {
    
    
public:
    vector<vector<int>> res;
    vector<int> path;

    void dfs(int k,int n,int start,int sum)
    {
    
    
        if(sum>n) return ;//剪枝
        if(path.size()==k) 
        {
    
    
            if(sum==n)
            res.push_back(path);
            return;
        }
        for(int i=start;i<=9-(k-path.size())+1;i++)//剪枝
        {
    
    
            
            path.push_back(i);
            sum+=i;
            dfs(k,n,i+1,sum);
            sum-=i;
            path.pop_back();
        }
    }

    vector<vector<int>> combinationSum3(int k, int n) {
    
    
        dfs(k,n,1,0);
        return res;
    }
};

思路参考来自–代码随想录