class Solution {
private:
vector<vector<int>> res;
vector<int> path;
void backtracking(int n, int k, int start) {
if (path.size() == k) {
res.push_back(path);
return;
}
for (int i = start; i <= n; i++) {
path.push_back(i);
backtracking(n, k, i + 1);
path.pop_back();
}
}
public:
vector<vector<int>> combine(int n, int k) {
path.clear();
res.clear();
backtracking(n, k, 1);
return res;
}
};
算法思想:首先得知道,组合和排序的区别,组合和无序的,排序是有序的。写回溯类型的题目,心中需要有一棵n叉树,这棵n叉树横向遍历是for循环,纵向遍历是递归,递归多少次是看自己需求并且是一个剪纸的过程。
class Solution {
private:
vector<vector<int>> res;
vector<int> path;
int Sum(vector<int> path) {
int sum = 0;
for (int i = 0; i < path.size(); i++) {
sum += path[i];
}
return sum;
}
void backtracking(int k, int n, int start) {
if (path.size() == k) {
if (Sum(path) == n) {
res.push_back(path);
return;
} else {
return;
}
}
for (int i = start; i <= 9 - (k - path.size()) + 1; i++) {
path.push_back(i);
backtracking(k, n, i + 1);
path.pop_back();
}
}
public:
vector<vector<int>> combinationSum3(int k, int n) {
backtracking(k, n, 1);
return res;
}
};
在这第二题中,有两次剪枝的过程,一个就是在当总和大于n时的一次剪枝,一个就是在要取得元素不够时的一次剪枝,这样使回溯算法相较于暴力解法更加快一些。