#216 10min 自己只想到了数值的剪枝没想到数量的了,看了随想录补上
几个注意的点:
1.记得for loop是从startIdx开始
2.用min的话,里面类型都得是int,但是size()是size_t所以需要转化
3.以下代码剪枝是:如果当前选择的数字和已经超过了n
,或者已经选择了k
个数字但和还没有达到n
,那么就没有必要再继续搜索了。
vector<int> vec;
vector<vector<int>> res;
void backtrack(int n, int k, int startIdx, int sum){
if(vec.size()==k){
if(sum==n) res.push_back(vec);
return;
}
int end=std::min(n-sum,9-(k-static_cast<int>(vec.size()))+1);
for(int i=startIdx; i<=end;i++){
vec.push_back(i);
backtrack(n,k,i+1,sum+i);
vec.pop_back();
}
}
vector<vector<int>> combinationSum3(int k, int n) {
backtrack(n,k,1,0);
return res;
}
clear可以加一下:
#17 电话数字
我一开始准备这么写,有点麻烦,看了随想录的觉得很好,用了他的初始化
自己写的 20min左右
std::unordered_map<int, std::vector<char>> map = {
{2, {'a', 'b', 'c'}}, {3, {'d', 'e', 'f'}}, {4, {'d', 'e', 'f'}},
{5, {'d', 'e', 'f'}}, {6, {'d', 'e', 'f'}}, {7, {'d', 'e', 'f'}},
{8, {'d', 'e', 'f'}}, {9, {'d', 'e', 'f'}}};
const string letterMap[10] = {
"", // 0
"", // 1
"abc", // 2
"def", // 3
"ghi", // 4
"jkl", // 5
"mno", // 6
"pqrs", // 7
"tuv", // 8
"wxyz", // 9
};
string str="";
vector<string> res;
void backtrack(int idx, string digits){
if(digits.size()==0 ) return;
if(digits.size()==str.size()){
res.push_back(str);
return;
}
int digit_val=digits[idx]-'0';
for(int i=0;i<letterMap[digit_val].size();i++){
str.push_back(letterMap[digit_val][i]);
backtrack(idx+1,digits);
str.pop_back();
}
}
vector<string> letterCombinations(string digits) {
backtrack(0,digits);
return res;
}
初始化很聪明,直接digit对应上idx,0,1就直接赋值为空就可以不要了
做这道题翻了个错误:在backtrack开头一开始写的: if(idx>=digits.size() ||digits.size()==0 ) return; idx>=digits.size()这个条件不要判断,因为就算idx超出了也要进去,这样就能走到结束条件,要把str push进去,不然结果就都是空的了。
digit 的char转int也不要忘记
另外一开始以为这道题解决不了重复的,后来发现可以,因为ac ca算两个