总体思路:回溯+组合。和之前的排列回溯不同是,返回结果无需考虑顺序,只要有数字即可,所以递归不用交换位置,只需要把层数(count)减掉,加上新的数字即可。
class Solution {
public:
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;
}
void backtracking(vector<vector<int>>& ans, vector<int>& comb, int& count, int start, int n, int k) {
if (count == k) {
ans.push_back(comb);
return;
}
for (int i = start; i <= n; i++) { // 递归构造搜索树,每次先确定最上面的根节点,然后递归确定下面几层的。
comb[count++] = i;
backtracking(ans, comb, count, i + 1, n, k); // 每次从这个递归出来,其实就是包括i的所有结果都已经添加完成了。
--count; // 因为上面i在当前递归层已经添加完了,所以count--,为下一个i安排在当前层。
}
}
};