深度优先搜索(dfs)
回溯算法
题目详情
给定两个整数 n
和 k
,返回范围 [1, n]
中所有可能的 k
个数的组合。
你可以按 任何顺序 返回答案。
示例1:
输入:n = 4, k = 2
输出:
[
[2,4],
[3,4],
[2,3],
[1,2],
[1,3],
[1,4],
]
示例2:
输入:n = 1, k = 1
输出:[[1]]
leetcode第46道排列是同一类题:
leetcode-46.全排列
类似于排列问题,我们也可以进行回溯。排列回溯的是交换的位置,而组合回溯的是是否把
当前的数字加入结果中。
我的代码:
class Solution
{
public:
//二维vector存答案 comb存每组满足条件的组合数 count计数 从pos这个数开始找其他数(起点)
void backtracking(vector<vector<int>> &ans, vector<int> &comb, int &count, int pos, int n, int k)
{
if (count == k) //退出条件:找够k个的组合数了
{
ans.push_back(comb); //将找到的一个组合push入ans
return;
}
for(int i = pos; i <= n; ++i) //从pos到n找数(起点)
{
comb[count++] = i; //将遍历到的i加入comb这组中
backtracking(ans, comb, count, i+1, n, k); //更新起点i+1继续dfs
--count; //回溯,从而将新加入comb的数剔除
}
}
vector<vector<int>> combine(int n, int k)
{
vector<vector<int>> ans;
vector<int> comb(k, 0); // comb大小是确定的,k
int count = 0;
backtracking(ans, comb, count, 1, n, k); //从1开始dfs
return ans;
}
};
涉及知识点:
1.深度优先搜索(dfs)
深度优先搜索(depth-first seach,DFS)在搜索到一个新的节点时,立即对该新节点进行遍历;因此遍历需要用先入后出的栈来实现,也可以通过与栈等价的递归来实现。对于树结构而言,由于总是对新节点调用遍历,因此看起来是向着“深”的方向前进。
2.回溯法
回溯法(backtracking)是优先搜索的一种特殊情况,又称为试探法,常用于需要记录节点状
态的深度优先搜索。通常来说,排列、组合、选择类问题使用回溯法比较方便。
顾名思义,回溯法的核心是回溯。在搜索到某一节点的时候,如果我们发现目前的节点(及
其子节点)并不是需求目标时,我们回退到原来的节点继续搜索,并且把在目前节点修改的状态
还原。这样的好处是我们可以始终只对图的总状态进行修改,而非每次遍历时新建一个图来储存
状态。在具体的写法上,它与普通的深度优先搜索一样,都有 [修改当前节点状态]→[递归子节
点] 的步骤,只是多了回溯的步骤,变成了 [修改当前节点状态]→[递归子节点]→[回改当前节点
状态]。