77. 组合
题目链接:力扣(LeetCode)官网 - 全球极客挚爱的技术成长平台
文档讲解:代码随想录
视频讲解:带你学透回溯算法-组合问题(对应力扣题目:77.组合)| 回溯法精讲!_哔哩哔哩_bilibili
思路:本题是回溯法的经典题目。直接的解法当然是使用for循环,例如示例中k为2,很容易想到 用两个for循环,这样就可以输出 和示例中一样的结果。同时,本题可以使用剪枝优化,可以剪枝的地方就在递归中每一层的for循环所选择的起始位置。如果for循环选择的起始位置之后的元素个数 已经不足 我们需要的元素个数了,那么就没有必要搜索了。
C++代码:
class Solution {
public:
vector<int> path;
vector<vector<int>> result;
void backTrack(int n, int k, int start){
if(path.size() == k){
result.push_back(path);
return;
}
for(int i = start; i <= n; i++){
path.push_back(i);
backTrack(n, k, i + 1);
path.pop_back();
}
}
vector<vector<int>> combine(int n, int k) {
backTrack(n, k, 1);
return result;
}
};
//剪枝
class Solution {
public:
vector<int> path;
vector<vector<int>> result;
void backTrack(int n, int k, int start){
if(path.size() == k){
result.push_back(path);
return;
}
for(int i = start; i <= n - (k - path.size()) + 1; i++){
path.push_back(i);
backTrack(n, k, i + 1);
path.pop_back();
}
}
vector<vector<int>> combine(int n, int k) {
backTrack(n, k, 1);
return result;
}
};