回溯法理论基础:
回溯法一般可以解决如下几种问题:
- 组合问题:N个数里面按一定规则找出k个数的集合
- 切割问题:一个字符串按一定规则有几种切割方式
- 子集问题:一个N个数的集合里有多少符合条件的子集
- 排列问题:N个数按一定规则全排列,有几种排列方式
- 棋盘问题:N皇后,解数独等
回溯法解决的问题是在集合中递归查找子集,可以抽象为树形结构,集合的大小就构成了树的宽度,递归的深度就构成了树的深度。回溯三部曲:
- 回溯函数模板返回值以及参数
- 回溯函数终止条件
- 回溯搜索的遍历过程
参考链接:代码随想录-回溯算法理论基础
77. 组合
给定两个整数 n 和 k,返回范围 [1, n] 中所有可能的 k 个数的组合。
你可以按 任何顺序 返回答案。
题目链接:LeetCode77.组合
文档讲解:代码随想录LeetCode77.组合
题解
剪枝操作:在搜索遍历时,每层需要留下(k-path.size())个数,所以需要遍历到n-(k-path.size())+1
class Solution {
private:
vector<vector<int>> result;
vector<int> path;
void backtracking(int n, int k, int startIndex) {
if (path.size() == k) {
result.push_back(path);
return;
}
for (int i = startIndex; i <= n - (k - path.size()) + 1; i++) {
path.push_back(i);
backtracking(n, k, i + 1);
path.pop_back();
}
}
public:
vector<vector<int>> combine(int n, int k) {
backtracking(n, k, 1);
return result;
}
};
216. 组合总和III
找出所有相加之和为 n 的 k 个数的组合,且满足下列条件:
- 只使用数字1到9
- 每个数字 最多使用一次
返回 所有可能的有效组合的列表 。该列表不能包含相同的组合两次,组合可以以任何顺序返回。
题目链接:LeetCode216.组合总和III
文档讲解:代码随想录LeetCode216.组合总和III
题解
剪枝操作:在搜索遍历时,每层需要留下(9-path.size())个数,所以需要遍历到n-(9-path.size())+1,同时注意回溯函数中当sum>n的时候即可return
class Solution {
public:
vector<vector<int>> result;
vector<int> path;
void backtracking(int k, int n, int sum, int startIndex) {
if (sum > n)
return;
if (path.size() == k) {
if (sum == n) {
result.push_back(path);
}
return;
}
for (int i = startIndex; i <= 9 - (k - path.size()) + 1; i++) {
sum += i;
path.push_back(i);
backtracking(k, n, sum, i + 1);
path.pop_back();
sum -= i;
}
}
vector<vector<int>> combinationSum3(int k, int n) {
backtracking(k, n, 0, 1);
return result;
}
};
17. 电话号码的字母组合
给定一个仅包含数字 2-9 的字符串,返回所有它能表示的字母组合。答案可以按 任意顺序 返回。
给出数字到字母的映射如下(与电话按键相同)。注意 1 不对应任何字母。
题解
建立数字到字母集的映射,再进行回溯
class Solution {
public:
const string lettersMap[10] = {"", "", "abc", "def", "ghi",
"jkl", "mno", "pqrs", "tuv", "wxyz"};
vector<string> result;
string str;
void backtracking(const string& digits, int index) {
if (index == digits.size()) {
result.push_back(str);
return;
}
int digit = digits[index] - '0';
string lettersDigit = lettersMap[digit];
for (int i = 0; i < lettersDigit.size(); i++) {
str.push_back(lettersDigit[i]);
backtracking(digits, index + 1);
str.pop_back();
}
}
vector<string> letterCombinations(string digits) {
if (digits.size() == 0)
return result;
backtracking(digits, 0);
return result;
}
};