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;
}
};