代码随想录算法训练营第22天 | 77.组合、216.组合总和Ⅲ、17.电话号码的字母组合

代码随想录算法训练营第22天 | 77.组合、216.组合总和Ⅲ、17.电话号码的字母组合

---

77.组合

题目链接：77.组合

解题思路

本题第一种解法是用 for 循环，示例中 k 为2，使用两个 for 循环即可解决。当 k=3 时，使用三层 for 循环解决，但当 k 为50时，此时使用 for 循环便会十分复杂。

本题考虑使用回溯算法，回溯三部曲如下：

• 确定回溯函数的参数与返回值：定义两个变量，一个用来存放符合条件的单一结果，另一个存放符合条件的所有结果集合。要从集合 n 中取 k 个数，那么 n 和 k 就是两个参数，此外还需要一个 startIndex，用来记录从哪里开始遍历集合的。
• 确定回溯函数的终止条件：当单一结果数组的大小达到 k，说明找到了一个组合
• 单层搜索的过程：for 循环横向遍历，从左到右取数，取过的数不再重复取。递归纵向遍历，如图所示。

代码实现

class Solution {
public:
    vector<vector<int>> result;
    vector<int> path;
    void backtracking(int n, int k, int startIndex) {
        if (path.size() == k) {
            result.push_back(path);
            return;
        }
        for (int i = startIndex; i <= n; i++) {
            path.push_back(i);
            backtracking(n, k, i + 1);
            path.pop_back();
        }
    }
    vector<vector<int>> combine(int n, int k) {
        result.clear();
        path.clear();
        backtracking(n, k, 1);
        return result;
    }
};

题目总结

将回溯法的搜索过程抽象为树形结构，可以直观地看出搜索的过程，用回溯的三部曲，可以逐步分析函数的参数、终止条件和单层搜索的逻辑。

216.组合总和Ⅲ

题目链接：216.组合总和Ⅲ

解题思路

本题和上一题的区别在于每种组合中不存在重复的数字。和上题一样用回溯来做：

• 函数参数和返回值：同样需要两个数组分别存放单一结果和结果集合，此外还需要参数：目标和 n、k、sum表示已经收集的元素的总和、startIndex。
• 终止条件：k个元素的限制
• 单层递归：path 收集每次选取的元素，sum 统计 path 中元素的总和。

代码实现

class Solution {
public:
    vector<vector<int>> result;
    vector<int> path;
    void backtracking(int targetSum, int k, int sum, int startIndex) {
        if (path.size() == k) {
            if (sum == targetSum) {
                result.push_back(path);
                return;
            }
        }

        for (int i = startIndex; i <= 9; i++) {
            sum += i;
            path.push_back(i);
            backtracking(targetSum, k, sum, i + 1);
            sum -= i;
            path.pop_back();
        }
    }

    vector<vector<int>> combinationSum3(int k, int n) {
        result.clear();
        path.clear();
        backtracking(n, k, 0, 1);
        return result;
    }
};

题目总结

在处理过程中有sum 加法操作，回溯过程就需要有减法操作，一一对应。

17.电话号码的字母组合

题目链接：17.电话号码的字母组合

解题思路

本题在使用回溯算法之前需要考虑：使用 map 或定义二维数组来实现数字和字母的映射。

• 确定递归函数参数：两个字符串，一个用于收集叶子节点的结果，另一个收集结果集合，此外还需要 index 记录遍历第几个数字
• 终止条件：index 等于输入的数字个数
• 单层递归：获取 index 指向的数字，找到对应的字符集，使用 for 循环处理这个字符集

代码实现

class Solution {
public:
    const string letterMap[10] = {
        "", "", "abc", "def", "ghi", "jkl", "mno", "pqrs", "tuv", "wxyz",
    };

    vector<string> result;
    string s;
    void backtracking(const string& digits, int index) {
        if (index == digits.size()) {
            result.push_back(s);
            return;
        }
        int digit = digits[index] - '0';
        string letters = letterMap[digit];
        for (int i = 0; i < letters.size(); i++) {
            s.push_back(letters[i]);
            backtracking(digits, index + 1);
            s.pop_back();
        }
    }

    vector<string> letterCombinations(string digits) {
        s.clear();
        result.clear();
        if (digits.size() == 0) {
            return result;
        }
        backtracking(digits, 0);
        return result;
    }
};

题目总结

注意：输入1 * #按键等等异常情况，在代码中可以加入对于这些异常情况的处理。