①、组合总和Ⅲ
找出所有相加之和为
n
的k
个数的组合,且满足下列条件:
- 只使用数字1到9
- 每个数字 最多使用一次
返回 所有可能的有效组合的列表 。该列表不能包含相同的组合两次,组合可以以任何顺序返回。
事例:
输入: k = 3, n = 7 输出: [[1,2,4]] 解释: 1 + 2 + 4 = 7 没有其他符合的组合了。
思路:
采用回溯算法,在组合长度等于k的时候判断是否为符合要求的组合(sum == n),当组合长度大于k时,就没必要继续添加了,直接return结束。
参数可以多增加一个curSum,记录当前组合tmp中的总和,节省时间。curSum需要和tmp操作一致,当tmp中加入某个数时,curSum加某个数,当tmp回溯时,curSum也要回溯。
代码:
public List<List<Integer>> combinationSum3(int k, int n) {
backtracking(k,n,1,0);
return res;
}
private List<List<Integer>> res = new ArrayList<>();
private LinkedList<Integer> tmp = new LinkedList<>();
public void backtracking(int k,int n,int startIndex,int curSum){
if(tmp.size() == k){
if(curSum == n){
//添加到res中
res.add(new ArrayList<>(tmp));
}
}
if(curSum > n){
//无需继续添加
return;
}
for(int i = startIndex;i <= n - (k - tmp.size()) + 1 && i <= 9;i++){
tmp.add(i);
curSum += i;
//递归
backtracking(k,n,i + 1,curSum);
//回溯
tmp.removeLast();
curSum -= i;
}
}
②、电话号码的字母组合
给定一个仅包含数字
2-9
的字符串,返回所有它能表示的字母组合。答案可以按 任意顺序 返回。给出数字到字母的映射如下(与电话按键相同)。注意 1 不对应任何字母。
事例:
输入:digits = "23" 输出:["ad","ae","af","bd","be","bf","cd","ce","cf"]
思路:
digits中的字符返回为2 ~ 9,每个字符对应的字母是固定的,故创建一个二维数组,作为对照表,采用回溯算法。
以往的回溯算法:
private List<List<Integer>> res = new ArrayList<>();
private LinkedList<Integer> tmp = new LinkedList<>();
for(int i = 1;i <= n;i++){
tmp.add(i);
for(int j = i + 1;j <= n;j++){
tmp.add(j);
res.add(new ArrayList<>(tmp));
tmp.removeLast();
}
tmp.removeLast();
}
for循环中的每层递归回溯代表上层N叉树从左到右的每个分支,电话号码的字母组合要概括所有数字对应的字符集组合,故只要for循环中i = 0的递归回溯,需要在for循环最后直接break,每个数字都有对应的字符集,故需要内嵌一个for循环遍历字符集,在取到对应的字符后再进行递归回溯。
如 "23":
代码:
public List<String> letterCombinations(String digits) {
if(digits == null || digits.length() == 0) return res;
List<char[]> help = new ArrayList<>();
help.add(new char[]{});
help.add(new char[]{});
help.add(new char[]{'a','b','c'});
help.add(new char[]{'d','e','f'});
help.add(new char[]{'g','h','i'});
help.add(new char[]{'j','k','l'});
help.add(new char[]{'m','n','o'});
help.add(new char[]{'p','q','r','s'});
help.add(new char[]{'t','u','v'});
help.add(new char[]{'w','x','y','z'});
backtracking(digits,0,help);
return res;
}
private List<String> res = new ArrayList<>();
StringBuilder sb = new StringBuilder();
public void backtracking(String digits,int startIndex,List<char[]> help){
if(startIndex >= digits.length()){
res.add(new String(sb));
}
for(int i = startIndex;i < digits.length();i++){
//取出对应字符
char c = digits.charAt(i);
int num = Integer.valueOf(c - '0');
char[] arr1 = help.get(num);
for(int j = 0;j < arr1.length;j++){
char tmp = arr1[j];
sb.append(tmp);
//递归
backtracking(digits,i + 1,help);
//回溯
sb.setLength(sb.length() - 1);
}
break;
}
}