这题的主要思路参考 75. Combination Sum, 大同小异,涉及的思想,依旧是 DFS。
要额外注意的有两点:
对于第一点,我们可以在递归中,给下一个 start
传参为 i+1 来解决。
对于第二点,有两种思路:
std::set<vector<int>>
来存储返回值,缺点是效率较低。if (i > start && num[i] == num[i-1]) continue;
无比轻巧的避过了重复项。我当然选择了第二种,效率遥遥领先呢。
#include <vector>
#include <set>
using std::vector;
#include <algorithm>
#include <functional>
class Solution {
public:
void dfs(const vector<int> &num, vector<vector<int>> &ret, int target, vector<int> cur, size_t start) {
if (target == 0) { ret.push_back(cur); return; }
for (auto i = start; i < num.size(); ++i)
if (i > start && num[i] == num[i-1]) continue;
else if (num.at(i) <= target) {
cur.push_back(num.at(i));
dfs(num, ret, target - num.at(i), cur, i+1);
cur.pop_back();
} else break;
}
vector<vector<int> > combinationSum2(vector<int> &num, int target) {
vector<vector<int>> ret;
std::sort(num.begin(), num.end());
dfs(num, ret, target, vector<int>{}, 0);
return ret;
}
};