给定n个不同的正整数,整数k(1<= k <= n)以及一个目标数字。
在这n个数里面找出K个数,使得这K个数的和等于目标数字,你需要找出所有满足要求的方案。
样例
给出[1,2,3,4],k=2, target=5,返回 [[1,4],[2,3]]
思路:
递归的方式找出所有可能的k个数的组合,符合要求的存入结果中
vector<vector<int> > kSumII(vector<int> A, int k, int target) {
// write your code here
vector<vector<int> > r;
sub(A, k, 0, 0, target, r);
return r;
}
void sub(vector<int>& A, int k, int pos, int cur, int target, vector<vector<int> >& r) {
static vector<int> tmp;
if(cur > target) return;
if(k == 0) {
if(cur == target) {
r.push_back(tmp);
}
return;
}
k -= 1;
for(; pos + k < A.size(); ++pos) {
tmp.push_back(A[pos]);
sub(A, k, pos + 1, cur + A[pos], target, r);
tmp.pop_back();
}
}