系列文章目录
前言
回溯法中递归函数参数很难一次性确定下来,一般先写逻辑,需要啥参数了,填什么参数。
216.组合总和III
Source: 题目
Note:和上一道题组合很像,不一样在于要记录每一层的sum,剪枝操作也不同,不过还是比较好想的
class Solution {
private:
vector<vector<int>> result;
vector<int> path;
public:
void backtracing(int targetSum, int k, int sum, int startIndex) {
//剪枝, 如果和超过就跳过
if (sum > targetSum) {
return;
}
// 终止条件
if (path.size() == k) {
// 仅当满足条件时放入结果
if (sum == targetSum) {
result.push_back(path);
}
return;
}
for (int i = startIndex; i <= 9 - (k - path.size()) + 1; i++) {
sum = sum + i;
path.push_back(i);
backtracing(targetSum, k, sum, i + 1);
// 回溯,sum变量也要回溯
sum = sum - i;
path.pop_back();
}
}
vector<vector<int>> combinationSum3(int k, int n) {
backtracing(n, k, 0, 1);
return result;
}
};
Tips:
17.电话号码的字母组合
Source: 题目
Note:题目给定一个键盘,和一串号码,要求输出所有可能的代表字符
关键在于:
建立letterMap、回溯函数参数选取、每一层集合的转换逻辑以及遍历逻辑
class Solution {
private:
vector<string> result;
string s;
const string letterMap[10] = {
"", // 0
"", // 1
"abc", // 2
"def", // 3
"ghi", // 4
"jkl", // 5
"mno", // 6
"pqrs", // 7
"tuv", // 8
"wxyz", // 9
};
public:
// index用于标记遍历到第几个数字
void backtracing(const string& digits, int index) {
if (index == digits.size()) {
result.push_back(s);
return;
}
// 将字符转换为int
int digit = digits[index] - '0';
// 获取按键的集合数字
string letters = letterMap[digit];
for (int i = 0; i < letters.size(); i++) {
s.push_back(letters[i]);
backtracing(digits, index + 1);
s.pop_back();
}
}
vector<string> letterCombinations(string digits) {
if (digits.size() == 0) {
return result;
}
backtracing(digits, 0);
return result;
}
};
Tips:
总结
加深了对回溯法的理解,没有想到可以建立一个map的方式进行解决。