刷题第25天 | 216.组合总和III、17.电话号码的字母组合

216. Combination Sum III

题目链接：216. Combination Sum III
思路链接：代码随想录回溯算法-组合总和III

思路

这题的思路与leetcode 77差不多。

将回溯过程抽象化成树状图之后，可以看到递归深度对应的是当前已经选取的元素的个数，因此只要当当前选取的元素个数与k相等时，就要结束递归。然后这道题要求的是选取的所有元素之和为n的组合是valid的，因此无非就是在结束条件内再增加一个if判断，如果符合，就在result中记录下来符合要求的组合。另外为了使计算元素之和与递归同时进行，在leetcode 77的基础上（需要处理的参数有k， n，startIndex），还需要再增加一个sum来记录当前递归的元素之和。另外还需注意通过剪枝操作来提升代码效率。

Code

import java.util.List;
import java.util.LinkedList;

class Solution {
    private List<List<Integer>> result = new LinkedList<>();
    private List<Integer> path = new LinkedList<>();
    
    private void backtracking(int k, int n, int startIndex, int sum) {
        // 剪枝操作
        if (sum > n) {
            return ;
        }
        // 结束条件，说明找到了含有k个元素的组合
        // 通过树状图可知，path.size()的最大值就是我们的递归深度，因此可以得到结束条件的内容
        if (path.size() == k) {
            // 找到valid组合
            if (sum == n) {
                result.add(new LinkedList<>(path));
            }
            return ;
        }
        // 遍历1-9中所有的元素
        // 剪枝操作
        for (int i = startIndex; i <= 9 - (k - path.size()) + 1; i++) {
            sum += i;
            path.add(i);
            backtracking(k, n, i + 1, sum);
            // 回溯操作
            path.remove(path.size() - 1);
            sum -= i;
        }
    }
    
    public List<List<Integer>> combinationSum3(int k, int n) {
        // 结束条件，说明找到了含有k个元素的组合
        backtracking(k, n, 1, 0);
        return result;
    }
}

17. Letter Combinations of a Phone Number

题目链接：17. Letter Combinations of a Phone Number
思路链接：代码随想录回溯算法-电话号码的字母组合

思路

我们首先要建立数字与字母的映射，我这里是用了String[],可以通过数字也就是数组下标来得到相应的字母字符串。
这题思路也可以抽象化成树形结构。可以看出，递归深度是取决于当前选取的元素的个数，也就是说，当我们选取的元素个数等于所需长度时，也就是我们遍历完digits时，递归需要结束。因此就使用index参数来记录当前遍历到的digit的index，如果index等于digits.length()时，结束递归。在递归操作步骤时，先从映射中得到，digit对应的字符串，然后对字符串进行遍历，在循环内进行递归操作，然后进行回溯操作。

Code

import java.util.List;
import java.util.LinkedList;

class Solution {
    private List<String> result = new LinkedList<>();
    private StringBuilder path = new StringBuilder();
    
    private void backtracking(String digits, String[] map, int index) {
        // 结束条件：index已经超过了digits的最后一位
        if (index == digits.length()) {
            result.add(path.toString());
            return ;
        }
        String str = map[digits.charAt(index) - '0'];
        for (int i = 0; i < str.length(); i++) {
            path.append(str.charAt(i));
            backtracking(digits, map, index + 1);
            // 回溯操作
            path.deleteCharAt(path.length() - 1);
        }
    }
    
    public List<String> letterCombinations(String digits) {
        if (digits == null || digits.length() == 0) {
            return result;
        }
        String[] map = {"", "", "abc", "def", "ghi", "jkl", "mno", "pqrs", "tuv", "wxyz"};
        backtracking(digits, map, 0);
        return result;
    }
}