原题
LeetCode 39. 组合总和
思路
- 根据题意我们需要找到数组中数组组合达到target。我们需要对candidates排个序,因为若不排序的话,不知道是否应该继续取下一个数。我们可以一直取最小的数,直至cnt>=target。当cnt刚好等于target时,ans加入目前的数字组合,若大于target,就可以退出该循环,倒退回前一个递归进行判断。因为你小的数都大于target了,比它更大的数肯定不会满足。
- 这里注意的时,如果我们直接search(candidates, 0, target); 相当于这个数字组合是以candidates[0]开头,当退至没有candidates[0]的时候递归就结束了,所以我们应该用一个循环来让所有数字都开头递归。
代码
class Solution {
public:
vector<int> tmp;
int cnt = 0;
vector<vector<int>> ans;
void search(vector<int>& candidates, int i, int target) {
if (i >= candidates.size()) return;
tmp.push_back(candidates[i]);
cnt += candidates[i];
if (cnt == target) ans.push_back(tmp);
else if (cnt < target) {
for(int j = i; j < candidates.size() && cnt + candidates[j] <= target; j++) {
search(candidates, j, target);
}
}
tmp.pop_back();
cnt -= candidates[i];
return;
}
vector<vector<int>> combinationSum(vector<int>& candidates, int target) {
sort(candidates.begin(), candidates.end());
for(int i = 0; i < candidates.size(); i++) search(candidates, i, target);
return ans;
}
};
运行截图
![在这里插入图片描述](https://i-blog.csdnimg.cn/blog_migrate/16d707cff71cdc692e99f9cc1cc0d619.png)
收获