组合
如果是求两两的组合。
可以写成:
for(int i=1;i<n;i++){
for(int j=i+1;j<n;j++){
cout<<i<<" "<<j;
}
}
三个的组合可以再套一层for循环,但是面对求k个组合时,不停的写for循环就太麻烦了。
考虑借用递归来解决for循环的嵌套,k个for循环就是递归k层,终止条件就是组合中有k个数。
对于二叉树来说,横向只用取root->left和root->right,纵向用递归解决。而现在是一个n叉树,所以用for循环从左往右取,纵向仍然用递归解决.
if(path.size()==k){
result.push_back(path);
return;
}
for(int i=startindex;i<=n;i++)//从左到右遍历宽度,相当于二叉树的先取left后取right
{
path.push_back(i);
combination(n,k,i+1);
path.pop_back();//回溯
//在combi(n,K,3)中才return回i=2的地方,然后弹出(1,2)中的末尾,i变成3,变成1,3;
}
}
剪枝优化
如果要求n=k=4时候,面对第一层的2,3,4,就没有遍历的必要了。
还需要添加k-path.size()个,剩下的个数是n-i+1个(因为i也算进去,比如i=2,n=3就剩下两个),比如满足n-i+1》k-path.size(),
一般剪枝都是在for循环i的范围上做文章。
39. 组合总和
和上一题是基本一样的,终止条件变为1.:sum>targetsum 2. path.size()==k 这时再判断sum和目标值是否相等。
还是一样要注意剪枝的操作。
17.电话号码的字母组合
题目
这题写了三十多分钟总算写出来了。
这题和之前不一样的地方就是它不只是在一个集合中组合,而是在多个集合中。
class Solution {
public:
vector<string> result;//最终结果
string path;//记录组合
map<char,string> mp;
void initialize_map(){//用哈希表映射出每一个号码对应的字符串字母
mp['2']="abc";
mp['3']="def";
mp['4']="ghi";
mp['5']="jkl";
mp['6']="mno";
mp['7']="pqrs";
mp['8']="tuv";
mp['9']="wxyz";
}
void backtracking(string digits,int startindex){//回溯
//startindex表示的是digits的下标
if(digits.size()==0)
return;//是空的话直接返回
if(path.size()==digits.size()){//当组合中的个数和digits长度相等时是递归重点
result.push_back(path);//记录组合
return ;
}
string temp=mp[digits[startindex]];
//记录下来当前号码对应的字符串,eg:‘2’对应的temp 是“abc”
for(int i=0;i<temp.size();i++){//temp 从左向右依次遍历
path.push_back(temp[i]);
backtracking(digits,startindex+1);//传入下一个字符串,在其中得到新的temp
path.pop_back();//回溯
}
}
vector<string> letterCombinations(string digits) {
int startindex=0;
initialize_map();
backtracking(digits,startindex);
return result;
}
};