前几天4.27回杭州,准备5月3号去上海梅龙镇广场办F-1面签。在家闲着无聊,代码状态又还不错,LC基本每天6,7道的进度,而且很多都是一次AC,一眨眼已经完成165道了。LC216很简单,很直接的回溯,但是之前写了一个Combination Sum II,感觉既然是一个系列的那就把这道也写一下吧。另外,我感觉我这样顺序做题有一个好处,就是在做到Combination Sum III的时候可以回看复习一下前些日子做的Combination Sum I & II,这样等于是在边刷题边复习。
Find all possible combinations of k numbers that add up to a number n, given that only numbers from 1 to 9 can be used and each combination should be a unique set of numbers.
Example 1:
Input: k = 3, n = 7
Output:
[[1,2,4]]
Example 2:
Input: k = 3, n = 9
Output:
[[1,2,6], [1,3,5], [2,3,4]]
这道题和之前的系列基本没有区别,多了一个限制,那就是必须是长度为K的组合,并且没有candidates数字,转而用1~9,也没有什么duplicate的限制,感觉比Combination Sum II 还简单一些。直接递归、回溯,没什么好说的,
void getCombination(vector<vector<int>>& res, vector<int>& cur, int k, int n, int start) {
if (k == 0 && n == 0) {
res.push_back(cur);
return;
}
if (n < 0) {
return;
}
for (int i = start; i <= 9; i++) {
cur.push_back(i);
getCombination(res, cur, k - 1, n - i, i + 1);
cur.pop_back();
}
}
vector<vector<int>> combinationSum3(int k, int n) {
vector<vector<int>> res;
vector<int> cur;
getCombination(res, cur, k, n, 1);
return res;
}