1、回溯
我们可以使用回溯进行遍历组合。我们首先利用二维数组ans
记录所有结果,利用comb
记录当前的组合,而后我们开始进行遍历。若当前组合中数字的个数恰好为k
,我们直接将当前组合加入ans
中并返回即可。否则我们在[pos,n]
的区间进行循环并不断向记录当前组合的comb
中添加元素,并在递归调用之后--count
。
class Solution {
public:
// 辅函数
void backtracking(vector<vector<int>> &ans, vector<int> &comb, int &count, int pos, int n, int k) {
if (count == k) {
ans.push_back(comb);
return;
}
for (int i = pos; i <= n; ++i) {
comb[count++] = i; // 修改当前节点状态
backtracking(ans, comb, count, i + 1, n, k); // 递归子节点
--count; // 回改当前节点状态
}
}
// 主函数
vector<vector<int>> combine(int n, int k) {
vector<vector<int>> ans;
vector<int> comb(k, 0);
int count = 0;
backtracking(ans, comb, count, 1, n, k);
return ans;
}
};