216.组合总和III
题目链接/文章讲解: 代码随想录
class Solution {
public:
vector<int> path;
vector<vector<int>> result;
int sum = 0;
void backTracking(int k, int n, int start) {
if (path.size() == k) {
if (sum == n)
result.push_back(path);
return;
}
for (int i = start; i <= 9; i++) {
path.push_back(i);
sum+=i;
backTracking(k, n, i+1);
path.pop_back();
sum-=i;
}
}
vector<vector<int>> combinationSum3(int k, int n) {
backTracking(k, n, 1);
return result;
}
};
我的做法,与组合总和类似,只不过加入了一个sum用来记录当前路径的总和。在终止条件处又添加了对总和的判断,如果等于n就将该路径加入result中。在遍历过程中,要加上sum+=i;和sum-=i;来进行对总和的记录。
也可以进行剪枝操作,具体代码如下:
void backtracking(int targetSum, int k, int sum, int startIndex) {
if (sum > targetSum) { // 剪枝操作
return;
}
if (path.size() == k) {
if (sum == targetSum) result.push_back(path);
return; // 如果path.size() == k 但sum != targetSum 直接返回
}
for (int i = startIndex; i <= 9 - (k - path.size()) + 1; i++) { // 剪枝
sum += i; // 处理
path.push_back(i); // 处理
backtracking(targetSum, k, sum, i + 1); // 注意i+1调整startIndex
sum -= i; // 回溯
path.pop_back(); // 回溯
}
}
首先if (sum > targetSum) { // 剪枝操作
return;
}操作属于终止条件,如果当前点和已经大于要求的和,直接返回。
其次for (int i = startIndex; i <= 9 - (k - path.size()) + 1; i++),在这里将i的值限定在能够满足k个元素的值当中,如果已经满足不了k个元素了,就不需要进行遍历了。
17.电话号码的字母组合
题目链接/文章讲解: 代码随想录
class Solution {
private:
const string letterMap[10] = {
"", // 0
"", // 1
"abc", // 2
"def", // 3
"ghi", // 4
"jkl", // 5
"mno", // 6
"pqrs", // 7
"tuv", // 8
"wxyz", // 9
};
public:
vector<string> result;
void getCombinations(const string& digits, int index, const string& s) { // 注意参数的不同
if (index == digits.size()) {
result.push_back(s);
return;
}
int digit = digits[index] - '0';
string letters = letterMap[digit];
for (int i = 0; i < letters.size(); i++) {
getCombinations(digits, index + 1, s + letters[i]); // 注意这里的不同
}
}
vector<string> letterCombinations(string digits) {
result.clear();
if (digits.size() == 0) {
return result;
}
getCombinations(digits, 0, "");
return result;
}
};
首先要使用一个letterMap数组来将数字映射为字母。之后在回溯函数中,有三个参数,digits值得是传入的电话号码;index指的是取第几个数字;s指的是累加的字符串,在每次递归中会增加在该次循环中得到的字母。
确定了参数后,终止条件就是简单的判断index的大小。在处理当前节点时,首先要取得当前数字,然后要根据当前数字来映射为字母,在循环中遍历每个数字所映射的字母,在递归中遍历所有数字,这就是这道回溯的树形结构。
同时在递归函数中,index + 1, s + letters[i]这两个参数也就是进行了隐式的回溯。