回溯的相关题目,有点难度。
1.力扣216(组合III)
本题基于昨天的组合问题,只是把终止条件进行了改变,当数组长度为长度为k时,并且和为n时才将路径存进结果集中,其他步骤和上个题一样。
List<List<Integer>> res =new ArrayList();
LinkedList<Integer> path = new LinkedList();
int sum=0;
public List<List<Integer>> combinationSum3(int k, int n) {
combination(k,n,1);
return res;
}
public void combination(int k, int n,int startIndex) {
if(path.size()==k&&sum==n){
res.add(new ArrayList(path));
return;
}
for(int i=startIndex;i<=9;i++){
path.add(i);
sum+=i;
combination(k,n,i+1);
path.removeLast();
sum-=i;
}
}
2.力扣17(电话号码的组合)
本题我们先使用numString数组来存储每个数字代码的字母,digits存储的是待遍历输入的元素,我们在定义一个sum来记录遍历的深度,numString【digits【sum】-‘0’】来取出代表的元素,然后我们利用回溯的模板进行编写,有一点需要改进,也就是我们直接创建个StringBuilder来动态存储字符串,这样节省空间。下面是回溯三部曲:
- 确定回溯函数参数:首先需要一个可变字符串temp来收集叶子节点的结果,然后用一个集合res保存起来,这两个变量依然定义为全局。这个sum是记录遍历第几个数字了,就是用来遍历digits的(题目中给出数字字符串),同时sum也表示树的深度。
public void letter(String digits,String[] numString,int sum)
- 确定终止条件:例如输入用例"23",两个数字,那么根节点往下递归两层就可以了,叶子节点就是要收集的结果集。那么终止条件就是如果index 等于 输入的数字个数(digits.size)了(本来sum就是用来遍历digits的),然后收集结果,结束本层递归。
if(sum==digits.length()){
res.add(temp.toString());
return;
}
- 确定单层遍历逻辑:首先要取sum指向的数字,并找到对应的字符集(手机键盘的字符集)。
String str = numString[digits.charAt(sum)-'0'];
for(int i=0;i<str.length();i++){
temp.append(str.charAt(i));
letter(digits,numString,sum+1);
temp.deleteCharAt(temp.length()-1);
}
整体代码:
List<String> res =new ArrayList();
StringBuilder temp = new StringBuilder();
public List<String> letterCombinations(String digits) {
if(digits==null||digits.length()==0){
return res;
}
String[] numString = {"", "", "abc", "def", "ghi", "jkl", "mno", "pqrs", "tuv", "wxyz"};
letter(digits,numString,0);
return res;
}
public void letter(String digits,String[] numString,int sum) {
if(sum==digits.length()){
res.add(temp.toString());
return;
}
String str = numString[digits.charAt(sum)-'0'];
for(int i=0;i<str.length();i++){
temp.append(str.charAt(i));
letter(digits,numString,sum+1);
temp.deleteCharAt(temp.length()-1);
}
}