文章目录
回溯模板
for循环可以理解是横向遍历,backtracking(递归)就是纵向遍历,这样就把这棵树全遍历完了,一般来说,搜索叶子节点就是找的其中一个结果了。
回溯和递归是相辅相成的。回溯法的效率,回溯法其实就是暴力查找,并不是什么高效的算法。
回溯法解决的问题都可以抽象为树形结构(N叉树)
回溯法可以解决几类问题,可以看出每一类问题都不简单。
组合问题:N个数里面按一定规则找出k个数的集合
切割问题:一个字符串按一定规则有几种切割方式
子集问题:一个N个数的集合里有多少符合条件的子集
排列问题:N个数按一定规则全排列,有几种排列方式
棋盘问题:N皇后,解数独等等
void backtracking(参数) {
if (终止条件) {
存放结果;
return;
}
for (选择:本层集合中元素(树中节点孩子的数量就是集合的大小)) {
处理节点;
backtracking(路径,选择列表); // 递归
回溯,撤销处理结果
}
}
17. 电话号码的字母组合
给定一个仅包含数字 2-9 的字符串,返回所有它能表示的字母组合。
给出数字到字母的映射如下(与电话按键相同)。注意 1 不对应任何字母。
示例: 输入:“23” 输出:[“ad”, “ae”, “af”, “bd”, “be”, “bf”, “cd”, “ce”, “cf”].
说明:尽管上面的答案是按字典序排列的,但是你可以任意选择答案输出的顺序。
class Solution {
private:
vector<string> result;
string path;
// wrong: int key[10]
string key[10] = {
"",
"",
"abc",
"def",
"ghi",
"jkl",
"mno",
"pqrs",
"tuv",
"wxyz"
};
// 键入的键盘数字 字符串, 字符串的下标
void backtracking(string inputDigits, int index) {
// 结束条件
if (path.size() == inputDigits.size()) {
result.push_back(path);
return;
}
// 获取下标为index 的 输入字符串所代表的数字
// wrong:int digit = inputDigits[index]
int digit = inputDigits[index] - '0';
// 获取数字所代表的的字符串
string letter = key[digit];
// 处理逻辑
for (int i = 0; i < letter.size(); i++) {
// 循环获取index所代表的的字母
path.push_back(letter[i]);
// 获取下一个index所代表的的字母
backtracking(inputDigits, index + 1);
path.pop_back();
}
}
public:
vector<string> letterCombinations(string digits) {
result.clear();
// wrong: 未处理空的情况
if (digits.size() == 0)
return result;
backtracking(digits, 0);
return result;
}
};
131. 分割回文串
给你一个字符串 s,请你将 s 分割成一些子串,使每个子串都是 回文串 。返回 s 所有可能的分割方案。
回文串 是正着读和反着读都一样的字符串。
示例 1:
输入:s = “aab” 输出:[[“a”,“a”,“b”],[“aa”,“b”]] 示例 2:
输入:s = “a” 输出:[[“a”]]
提示:
1 <= s.length <= 16 s 仅由小写英文字母组成
来源:力扣(LeetCode)
链接:https://leetcode-cn.com/problems/palindrome-partitioning
著作权归领扣网络所有。商业转载请联系官方授权,非商业转载请注明出处。
class Solution {
private:
vector<vector<string>> result;
// wrong: vector<int> path;
vector<string> path;
// 判断回文与否 [begin, end]
bool isHuiWen(string s, int begin, int end) {
for (int i = begin, j = end; i < j; i++, j--) {
if (s[i] != s[j])
return false;
}
return true;
}
void backtracking(string s, int startIndex) {
// !!终止条件,起始位置
if (startIndex >= s.size()) {
result.push_back(path);
return;
}
// 处理逻辑,
for (int i = startIndex; i < s.size(); i++) {
// 判断是否是回文
if (isHuiWen(s, startIndex, i)) {
// !!截取从startIndex开始,长度为startIndex - i + 1的子串加入到path
// wrong: string sub = s.substr(startIndex, startIndex - i + 1);
string sub = s.substr(startIndex, i - startIndex + 1);
path.push_back(sub);
} else {
continue;
}
// !!起始位置从i + 1开始
backtracking(s, i + 1);
path.pop_back();
}
}
public:
vector<vector<string>> partition(string s) {
result.clear();
path.clear();
backtracking(s, 0);
return result;
}
};
93. 复原 IP 地址
有效 IP 地址 正好由四个整数(每个整数位于 0 到 255 之间组成,且不能含有前导 0),整数之间用 ‘.’ 分隔。
例如:“0.1.2.201” 和 “192.168.1.1” 是 有效 IP 地址,但是
“0.011.255.245”、“192.168.1.312” 和 “192.168@1.1” 是 无效 IP 地址。
给定一个只包含数字的字符串 s ,用以表示一个 IP 地址,返回所有可能的有效 IP 地址,这些地址可以通过在 s 中插入 ‘.’
来形成。你不能重新排序或删除 s 中的任何数字。你可以按 任何 顺序返回答案。来源:力扣(LeetCode)
链接:https://leetcode-cn.com/problems/restore-ip-addresses
著作权归领扣网络所有。商业转载请联系官方授权,非商业转载请注明出处。
class Solution {
private:
vector<string> result;
string path;
bool isValid(string s, int begin, int end) {
if (begin > end) {
return false;
}
//wrong:if (s[0] == "0" && begin != end) {
if (s[begin] == '0' && begin != end) {
return false;
}
int num = 0;
for (int i = begin; i <= end; i++) {
if (s[i] - '0' < 0 || s[i] - '0' > 9)
return false;
// int ip = s[i] - '0';
// wrong: num += (num * 10 + ip);
num = num * 10 + (s[i] - '0');
if (num > 255)
return false;
}
return true;
}
void backtracking(string& s, int startIndex, int piontNum) {
// 终止条件,点数为3
if (piontNum == 3) {
if (isValid(s, startIndex, s.size() - 1))
result.push_back(s);
return;
}
// 什么时候插入点?
for (int i = startIndex; i < s.size(); i++) {
if (isValid(s, startIndex, i)) {
//path.push_back(s[i]);
//path.push_back('.');
s.insert(s.begin() + i + 1, '.');
// }else {
// break;
// }
piontNum++;
backtracking(s, i + 2, piontNum);
// path.pop_back()
// path.pop_back()
s.erase(s.begin() + i + 1);
piontNum--;
}
else {
break;
}
}
}
public:
vector<string> restoreIpAddresses(string s) {
result.clear();
path.clear();
if (s.size() > 12) return result;
backtracking(s, 0, 0);
return result;
}
};
// 需要一个判断是否合法的方法
// 需要回溯分割的方法
78. 子集
给你一个整数数组 nums ,数组中的元素 互不相同 。返回该数组所有可能的子集(幂集)。
解集 不能 包含重复的子集。你可以按 任意顺序 返回解集。
示例 1:
输入:nums = [1,2,3] 输出:[[],[1],[2],[1,2],[3],[1,3],[2,3],[1,2,3]] 示例 2:
输入:nums = [0] 输出:[[],[0]]
来源:力扣(LeetCode) 链接:https://leetcode-cn.com/problems/subsets
著作权归领扣网络所有。商业转载请联系官方授权,非商业转载请注明出处。
class Solution {
private:
vector<vector<int>> result;
vector<int> path;
void backtracking(vector<int>& nums, int startIndex) {
// 遍历所有!!需要放在终止条件前
result.push_back(path);
if (startIndex > nums.size()) {
return ;
}
// 组合,从startIndex开始
for (int i = startIndex; i < nums.size(); i++) {
path.push_back(nums[i]);
backtracking(nums, i + 1);
path.pop_back();
}
}
public:
vector<vector<int>> subsets(vector<int>& nums) {
result.clear();
path.clear();
backtracking(nums, 0);
return result;
}
};
90. 子集 II
给你一个整数数组 nums ,其中可能包含重复元素,请你返回该数组所有可能的子集(幂集)。
解集 不能 包含重复的子集。返回的解集中,子集可以按 任意顺序 排列。
示例 1:
输入:nums = [1,2,2] 输出:[[],[1],[1,2],[1,2,2],[2],[2,2]] 示例 2:
输入:nums = [0] 输出:[[],[0]]
提示:
1 <= nums.length <= 10
-10 <= nums[i] <= 10来源:力扣(LeetCode) 链接:https://leetcode-cn.com/problems/subsets-ii
著作权归领扣网络所有。商业转载请联系官方授权,非商业转载请注明出处。
class Solution {
private:
vector<vector<int>> result;
vector<int> path;
void backtracking(vector<int>& nums, int startIndex) {
result.push_back(path);
for (int i = startIndex; i < nums.size(); i++) {
// !!! 层去重
if (i > startIndex && nums[i] == nums[i-1]) {
continue;
}
path.push_back(nums[i]);
backtracking(nums, i + 1);
path.pop_back();
}
}
public:
vector<vector<int>> subsetsWithDup(vector<int>& nums) {
result.clear();
path.clear();
// 去重要排序
sort(nums.begin(), nums.end());
backtracking(nums, 0);
return result;
}
};
491. 递增子序列
给你一个整数数组 nums ,找出并返回所有该数组中不同的递增子序列,递增子序列中 至少有两个元素 。你可以按 任意顺序 返回答案。
数组中可能含有重复元素,如出现两个整数相等,也可以视作递增序列的一种特殊情况。
示例 1:
输入:nums = [4,6,7,7]
输出:[[4,6],[4,6,7],[4,6,7,7],[4,7],[4,7,7],[6,7],[6,7,7],[7,7]] 示例 2:输入:nums = [4,4,3,2,1] 输出:[[4,4]]
来源:力扣(LeetCode)
链接:https://leetcode-cn.com/problems/increasing-subsequences
著作权归领扣网络所有。商业转载请联系官方授权,非商业转载请注明出处。
class Solution {
private:
vector<vector<int>> result;
vector<int> path;
void backtracking(vector<int>& nums, int startIndex) {
// 因为要遍历所有树枝,不能在此终止
//wrong:if (path.size() > 0) {
if (path.size() > 1) {
result.push_back(path);
}
unordered_set<int> uset;
for (int i = startIndex; i < nums.size(); i++) {
// 去重,且去除前大于后的
// wrong: 因为不能排序,所以不能通过这样来去重,if ((i > startIndex && nums[i] == nums[i-1]) || (path.size() > 0 && path.back() > nums[i])) {
if ((!path.empty() && path.back() > nums[i]) || uset.find(nums[i]) != uset.end()){
continue;
}
// 本层中不能重复,不需
uset.insert(nums[i]);
path.push_back(nums[i]);
backtracking(nums, i + 1);
path.pop_back();
}
}
public:
vector<vector<int>> findSubsequences(vector<int>& nums) {
result.clear();
path.clear();
backtracking(nums, 0);
return result;
}
};
小结
- 和子集问题很像,但是不同的是不能排序
- vector 末尾 back(),头front()
- unordered_set ,insert(),
46. 全排列
给定一个不含重复数字的数组 nums ,返回其 所有可能的全排列 。你可以 按任意顺序 返回答案。
示例 1:
输入:nums = [1,2,3] 输出:[[1,2,3],[1,3,2],[2,1,3],[2,3,1],[3,1,2],[3,2,1]]
示例 2:输入:nums = [0,1] 输出:[[0,1],[1,0]] 示例 3:
输入:nums = [1] 输出:[[1]]
提示:
1 <= nums.length <= 6
-10 <= nums[i] <= 10 nums 中的所有整数 互不相同来源:力扣(LeetCode) 链接:https://leetcode-cn.com/problems/permutations
著作权归领扣网络所有。商业转载请联系官方授权,非商业转载请注明出处。
class Solution {
private:
vector<vector<int>> result;
vector<int> path;
void backtracking(vector<int>& nums, vector<bool>& used) {
// 全排列,长度相等就加入到结果集
if (path.size() == nums.size()) {
result.push_back(path);
return;
}
for (int i = 0; i < nums.size(); i++) {
// path 中已经有了
if (used[i] == true) {
continue;
}
used[i] = true;
path.push_back(nums[i]);
backtracking(nums, used);
path.pop_back();
used[i] = false;
}
}
public:
vector<vector<int>> permute(vector<int>& nums) {
result.clear();
path.clear();
vector<bool> used(nums.size(), false);
backtracking(nums, used);
return result;
}
};
小结
- 排列问题,遍历时从0开始而不是startIndex
- 需要used数组记录path里都放了哪些元素了
47. 全排列 II
给定一个可包含重复数字的序列 nums ,按任意顺序 返回所有不重复的全排列。
示例 1:
输入:nums = [1,1,2] 输出: [[1,1,2], [1,2,1], [2,1,1]] 示例 2:
输入:nums = [1,2,3] 输出:[[1,2,3],[1,3,2],[2,1,3],[2,3,1],[3,1,2],[3,2,1]]
提示:
1 <= nums.length <= 8
-10 <= nums[i] <= 10来源:力扣(LeetCode) 链接:https://leetcode-cn.com/problems/permutations-ii
著作权归领扣网络所有。商业转载请联系官方授权,非商业转载请注明出处。
class Solution {
private:
vector<vector<int>> result;
vector<int> path;
void backtracking(vector<int>& nums, vector<bool>& used) {
if (path.size() == nums.size()) {
result.push_back(path);
return;
}
for (int i = 0; i < nums.size(); i++) {
// wrong: used[i] == false
if (i > 0 && nums[i] == nums[i-1] && used[i - 1] == false) {
continue;
}
if (used[i] == false) {
used[i] = true;
path.push_back(nums[i]);
backtracking(nums, used);
path.pop_back();
used[i] = false;
}
}
}
public:
vector<vector<int>> permuteUnique(vector<int>& nums) {
result.clear();
path.clear();
// wrong:忘了要排序了
sort(nums.begin(), nums.end());
vector<bool> used(nums.size(), false);
backtracking(nums, used);
return result;
}
};
小结
- used[i - 1] == true,说明同一树支nums[i - 1]使用过
- used[i - 1] == false,说明同一树层nums[i - 1]使用过
332. 重新安排行程
class Solution {
private:
vector<string> result;
unordered_map<string, map<string, int>> targets;
bool backtracking(int ticketNum) {
if (result.size() == ticketNum + 1) {
return true;
}
//wrong: for (const pair<string, int>& target : targets[result[result.size() - 1]]) {
//wrong: for (pair<string, int> target : targets[result[result.size() - 1]]) {
for (pair<const string, int>& target : targets[result[result.size() - 1]]) {
if (target.second > 0) {
result.push_back(target.first);
target.second--;
if (backtracking(ticketNum)) return true;
result.pop_back();
target.second++;
}
}
return false;
}
public:
vector<string> findItinerary(vector<vector<string>>& tickets) {
result.clear();
targets.clear();
for (const vector<string>& ticket : tickets) {
targets[ticket[0]][ticket[1]]++;
}
// wrong: 忘了起始
result.push_back("JFK"); // 起始机场
backtracking(tickets.size());
return result;
}
};
小结
for (pair<const string, int>& target : targets[result[result.size() - 1]])
pair里要有const,因为map中的key是不可修改的
51. N 皇后
class Solution {
private:
vector<vector<string>> result;
void backtracking(int n, int row, vector<string>& chessboard) {
// 终止
if (row == n) {
result.push_back(chessboard);
return;
}
for (int col = 0; col < n; col++) {
if (isValid(row, col, n, chessboard)) {
chessboard[row][col] = 'Q';
backtracking(n, row + 1, chessboard);
chessboard[row][col] = '.';
}
}
}
bool isValid(int row, int col, int n, vector<string>& chessboard) {
// 列
for (int i = 0; i < row; i++) {
if (chessboard[i][col] == 'Q') {
return false;
}
}
// 主对角线
for (int i = row - 1, j = col - 1; i >= 0 && j >= 0; j--, i-- ) {
if (chessboard[i][j] == 'Q') {
return false;
}
}
// 辅对角线
for (int i = row - 1, j = col + 1; i >= 0 && j < n; i-- , j++) {
if (chessboard[i][j] == 'Q') {
return false;
}
}
return true;
}
public:
vector<vector<string>> solveNQueens(int n) {
result.clear();
// 初始化!
vector<string> chessboard(n, string(n, '.'));
backtracking(n, 0, chessboard);
return result;
}
};