1.理论基础
题目链接/文章讲解:代码随想录
2.组合
代码:未优化版本
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; 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;
}
};
代码:剪枝优化版本
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;
}
};
note:这里剪枝的部分体现在for循环的循环条件的修改。
已经遍历收获的元素个数是path.size(),还需要收获k - path.size()个元素,总共有n个元素,因此我们需要的最大的开始位置是 n - (k - path.size()) + 1。
3.组合总和3
题目链接/文章讲解:代码随想录
视频讲解:和组合问题有啥区别?回溯算法如何剪枝?| LeetCode:216.组合总和III_哔哩哔哩_bilibili
代码:回溯法
class Solution {
private:
vector<int> path;
vector<vector<int>> result;
void backtracking(int k, int sum, int startIndex){
if(path.size() == k){
if(sum == 0){
result.push_back(path);
}
return;
}
for(int i = startIndex; i <= 9 - (k - path.size()) + 1; i++){
path.push_back(i);
sum -= i;
backtracking(k,sum,i + 1);
sum += i;
path.pop_back();
}
}
public:
vector<vector<int>> combinationSum3(int k, int n) {
backtracking(k,n,1);
return result;
}
};
代码:还可以做剪枝,如果当前的sum已经<0,就没必要递归下去了
class Solution {
private:
vector<int> path;
vector<vector<int>> result;
void backtracking(int k, int sum, int startIndex){
if(sum < 0) return;
if(path.size() == k){
if(sum == 0){
result.push_back(path);
}
return;
}
for(int i = startIndex; i <= 9 - (k - path.size()) + 1; i++){
path.push_back(i);
sum -= i;
backtracking(k,sum,i + 1);
sum += i;
path.pop_back();
}
}
public:
vector<vector<int>> combinationSum3(int k, int n) {
backtracking(k,n,1);
return result;
}
};
4.电话号码的字母组合
题目链接/文章讲解:代码随想录
代码:
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
};
vector<string> result;
string s;
void backtracking(string digits, int index){
if(index == digits.size()){
result.push_back(s);
return;
}
int digit = digits[index] - '0';
string letter = letterMap[digit];
for(int i = 0; i < letter.size(); i++){
s.push_back(letter[i]);
backtracking(digits,index + 1);
s.pop_back();
}
}
public:
vector<string> letterCombinations(string digits) {
if(digits.size() == 0){
return result;
}
backtracking(digits,0);
return result;
}
};
note:这里的技巧就是,首先得想到我们需要一个变量标识出我们从哪个对应的字符串来取值,也就是代码里的index。在每一次进行操作的时候,我们需要把当前的取值的字符串找到——因此就要去想已知的变量怎么转化从这个我们需要的字符串。找到index对应的digits,把它变成int类型,再根据这个整数,从我们建立的映射关系里找出当前需要取值的字符串就好了。