LeetCode 216. 组合总和 III
class Solution {
public:
vector<vector<int>> result;
vector<int> path;
int count=0;
void backtracking(int k,int n,int startindex){
if(path.size()==k && count==n){
result.push_back(path);
return ;
}
if(path.size()==k || count>n){
return ;
}
for(int i=startindex;i<=9;i++){
count+=i;
path.push_back(i);
backtracking(k,n,i+1);
count-=i;
path.pop_back();
}
}
vector<vector<int>> combinationSum3(int k, int n) {
backtracking(k,n,1);
return result;
}
};
注意的是count进行加减回溯的对象都是i
上述代码第二个if内的count>n,就是剪枝操作,避免了不必要的递归。
LeetCode 17.电话号码的字母组合
class Solution {
public:
string path;
vector<string> result;
string arr[10]={"0","0","abc","def","ghi","jkl","mno","pqrs","tuv","wxyz"};
void backtracking(string digits,int startindex){
if(path.size()==digits.size()){
result.push_back(path);
return ;
}
for(int i=startindex;i<digits.size();i++){
for(int j=0;j<arr[digits[i]-'0'].size();j++){
path+=arr[digits[i]-'0'][j];
backtracking(digits,i+1);
path.pop_back();
}
}
}
vector<string> letterCombinations(string digits) {
if(digits.empty())return result;
backtracking(digits,0);
return result;
}
};
我自己写出来的方法,哥们真他妈的牛逼吧。
本题也是利用回溯思想,再之前给出的回溯模板中进行再次变化,将每一个电话键盘上的数字字符与一段string相对应,具体是用arr数组来实现的,接着一一对应之后,再之前模板for中,再嵌套一个for循环,里面又加了一层对于选择具体字符的判断。我太他妈吊了。
这里面stratindex就是具体遍历到的电话上的哪一个数字位置。
刚刚又仔细思考了下,我这段代码好像外层的一个for循环根本就没有什么用,因为这个题目是输入的数字与返回的字符串长度一样的,那第一个数字的情况遍历完从第二个开始算起,根本就凑不齐所需要的长度。
我测,还是卡哥的代码更透彻。
class Solution {
public:
string path;
vector<string> result;
string arr[10]={"0","0","abc","def","ghi","jkl","mno","pqrs","tuv","wxyz"};
void backtracking(string digits,int startindex){
if(path.size()==digits.size()){
result.push_back(path);
return ;
}
for(int j=0;j<arr[digits[startindex]-'0'].size();j++){
path+=arr[digits[startindex]-'0'][j];
backtracking(digits,startindex+1);
path.pop_back();
}
}
vector<string> letterCombinations(string digits) {
if(digits.empty())return result;
backtracking(digits,0);
return result;
}
};
我试了下,确实是这样。。。