本文旨在记录解题过程中的收获,使用C++
初始化一维数组:vector<int> path
初始化二维数组: vector<vector<int>> result
解题思路:回溯法
从startIndex位置开始遍历,将该位置的元素压入中转栈,继续下一层递归,当栈里面的元素数量符合要求的数量的时候,就将中转栈的元素传到答案栈中。返回上一层函数,弹出中转栈的一个元素(这一步是关键,千万不要漏,体现了回溯的思想)
如何理解path.pop_back()的作用?
假设k为2,并且只存在一个正确的组合,当经过两次递归后,我们就可以找到一个符合要求的组合,把这个结果压入答案栈之后,就应该清空中转栈,为下一次寻找做准备。首次调用backTracking函数->压入一号元素,第一次递归->压入二号元素,第二次递归->符合组合条件,输出答案,返回第一次递归->弹出二号元素,退回首次backTracking->弹出一号元素,回溯完成。
我的代码:
class Solution {
public:
vector<int> path;//一维数组
vector<vector<int>> result;
vector<vector<int>> combine(int n, int k) {
backTracking(n,k,1);
return result;
}
void backTracking(int n,int k,int startIndex){
if(path.size()==k){
result.push_back(path);
return ;
}
for(int i=startIndex;i<=n;i++){
path.push_back(i);
backTracking(n,k,i+1);
path.pop_back();
}
}
};
官方代码:
class Solution {
public:
vector<vector<int>> combine(int n, int k) {
back(n,k,1);
return result;
}
private:
void back(int n,int k,int index){
if(path.size()==k){
result.push_back(path);
return;
}
for (int i=index;i<=n;i++){
path.push_back(i);
back(n,k,i+1);
path.pop_back();
}
}
vector<int> path;
vector<vector<int>> result;
};