链接
https://leetcode-cn.com/problems/combination-sum-iii/
耗时
解题:38 min
题解:5 min
题意
找出所有相加之和为 n 的 k 个数的组合。组合中只允许含有 1 - 9 的正整数,并且每种组合中不存在重复的数字。
说明:
- 所有数字都是正整数。
- 解集不能包含重复的组合。
思路
是这个 【leetcode】39. 组合总和(combination-sum)(回溯)[中等] 的变型。
相当于 candidates = {1, 2, 3, 4 , 5, 6, 7, 8, 9}; 目标数 target = n。只是加了一个限制,组合中的数字个数是固定的 k。
时间复杂度: O ( C 9 k ∗ k ) O(C_9^k*k) O(C9k∗k)
AC代码
class Solution {
private:
int k;
vector<int> seq;
vector<vector<int>> ans;
public:
void dfs(int pos, int rest) {
if(seq.size() == k) {
if(rest == 0) {
ans.push_back(seq);
}
return ;
}
for(int i = pos; i <= 9 && rest >= i; ++i) {
seq.push_back(i);
dfs(i+1, rest-i);
seq.pop_back();
}
}
vector<vector<int>> combinationSum3(int k, int n) {
this->k = k;
dfs(1, n);
return ans;
}
};