这两道题都是基于回溯的基本问题。
216.组合总和III
这道题是77.组合问题的变体,只不过终止条件多了一个和等于n。
class Solution {
List<List<Integer>> res=new ArrayList<>();
List<Integer> path=new ArrayList<>();
public List<List<Integer>> combinationSum3(int k, int n) {
res.clear();//如果多次调用函数要先清空结果
path.clear();
backtracking(k,n,1,0);
return res;
}
public void backtracking(int k,int n,int startIndex,int sum){
if(sum>n || path.size()>k) return;//剪枝条件
if(path.size()==k){
if(sum==n){
res.add(new ArrayList<>(path));
return;
}
}
for(int i=startIndex;i<=9;i++){
path.add(i);
backtracking(k,n,i+1,sum+i);
path.remove(path.size()-1);
}
}
}
时间复杂度: O(n * 2^n)
空间复杂度: O(n)
17.电话号码的字母组合
本题是多个集合求组合,所以在回溯的搜索过程中,都有一些细节需要注意的。
class Solution {
//设置全局列表存储最后的结果
List<String> list = new ArrayList<>();
public List<String> letterCombinations(String digits) {
if (digits == null || digits.length() == 0) {
return list;
}
//初始对应所有的数字,为了直接对应2-9,新增了两个无效的字符串""
String[] numString = {"", "", "abc", "def", "ghi", "jkl", "mno", "pqrs", "tuv", "wxyz"};
//迭代处理
backTracking(digits, numString, 0);
return list;
}
//每次迭代获取一个字符串,所以会设计大量的字符串拼接,所以这里选择更为高效的 StringBuild
StringBuilder temp = new StringBuilder();
//比如digits如果为"23",num 为0,则str表示2对应的 abc
public void backTracking(String digits, String[] numString, int num) {
//遍历全部一次记录一次得到的字符串
if (num == digits.length()) {
list.add(temp.toString());
return;
}
//str 表示当前num对应的字符串
String str = numString[digits.charAt(num) - '0'];
for (int i = 0; i < str.length(); i++) {
temp.append(str.charAt(i));
//c
backTracking(digits, numString, num + 1);//注意这里传入的是num+1,而不是i+1
//剔除末尾的继续尝试
temp.deleteCharAt(temp.length() - 1);
}
}
}
时间复杂度: O(3^m * 4^n),其中 m 是对应四个字母的数字个数,n 是对应三个字母的数字个数
空间复杂度: O(3^m * 4^n)