LeetCode每日一题 2020-9-11 216.组合总和III 回溯dfs 子集树 +剪枝

LeetCode每日一题 2020-9-11 216.组合总和III 回溯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]]

来源：力扣（LeetCode）
链接：https://leetcode-cn.com/problems/combination-sum-iii

解题思路
本题可以使用回溯子集树的方法来解决，子集树相当于对于每一个数选择加入或不加入。同时在搜索的过程中可以使用剪枝函数。最后需要注意的就是选择数字的个数，需要等于k。

题解代码

class Solution {
public:
   int leftSum[11]={0};
    void dfs(int t,int n,int k,int sum,vector<vector<int>>&ans,vector<int>& temp){
        if(sum==n){
            if(temp.size()==k){//正好是k个数
                ans.push_back(temp);
            }
            return;
        }
        if(sum+leftSum[t]<n){//剪枝
            return;
        }
        if(t>9){
            return;
        }
        temp.push_back(t);
        sum+=t;
        dfs(t+1,n,k,sum,ans,temp);
        temp.pop_back();
        sum-=t;
        dfs(t+1,n,k,sum,ans,temp);
    }
    vector<vector<int>> combinationSum3(int k, int n) {
        for(int i=9;i>0;i--){
            leftSum[i]+=(i+leftSum[i+1]);
        }
        vector<vector<int>> ans;
        vector<int> temp;
        dfs(1,n,k,0,ans,temp);
        return ans;
    }
};