代码随想录算法训练营第二十五天|LeetCode 216.组合总和III、17.电话号码的字母组合

LeetCode 216.组合总和III

题目链接

class Solution {
public:
    // 定义一个全局的vector保存所有符合条件的组合
    vector<vector<int> > results;
    vector<int> path; // 符合条件的结果
    // 回溯三部曲：
    // 1. 回溯函数返回值以及参数
    void backtracking(int k, int n, int startIndex, int sum) { // sum是数字组合的和
        // 2. 回溯函数终止条件：达到什么条件后需要结束递归
        if(path.size() == k) { // 如果path里的数字个数等于k一定return
            if(sum == n) { // 若此时的和等于n，则需要将符合条件的集合加入到results中
                results.push_back(path);
            }
            return;
        }
        // 3. 回溯搜索的遍历过程
        for(int i = startIndex; i <= 9 - (k - path.size()) + 1; i ++) {
            if(sum + i > n) { // 若当前和加上下一个数的和大于n，则直接退出循环，没有往后进行的必要
                return;
            }
            sum += i;
            // 将i加入到path中
            path.push_back(i);
            // 继续往下走
            backtracking(k, n, i + 1, sum);
            // 回溯
            sum -= i;
            path.pop_back();
        }
    }
    vector<vector<int>> combinationSum3(int k, int n) {    
        backtracking(k, n, 1, 0);
        return results;
    }
};

LeetCode 17.电话号码的字母组合

题目链接

class Solution {
public:
    // 二维数组保存数字对应字母
    const string letterMap[10] = {
        "", 
        "",
        "abc",
        "def",
        "ghi",
        "jkl",
        "mno",
        "pqrs",
        "tuv",
        "wxyz",
    };
    //  保存结果集合
    vector<string> results;
    string path;
    // 回溯三部曲
    // 1. 回溯函数返回值以及参数
    void backtracking(const string& digits, int startIndex) {
        // 2. 回溯函数终止条件
        // startIndex代表此次需要从字符串的那个位置开始，如果开始位置等于字符串的长度，则返回。
        if(startIndex == digits.size()) {
            results.push_back(path);
            return;
        }
        // 3. 回溯搜索的遍历过程
        int digit = digits[startIndex] - '0'; // 将字符串中字符转换成数字
        // 把对应的字符集取出来
        string letters = letterMap[digit];
        for(int i = 0; i < letters.size(); i ++) {
            path.push_back(letters[i]);
            backtracking(digits, startIndex + 1);
            // 回溯
            path.pop_back();
        }
    }
    vector<string> letterCombinations(string digits) {
        // 处理digits为空字符串时的情况
        if(digits == "") {
            return results;
        }
        backtracking(digits, 0);
        return results;
    }
};