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

Leetcode216.组合总和III

文章链接：代码随想录
题目链接：216.组合总和III

思路：比组合问题多加了一个条件，剪枝时同样可以多剪一处。
未剪枝：

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

        for (int i = startIndex; i <= 9; i++){
            sum += i;
            path.push_back(i);
            backtracking(k, n, sum, i + 1);
            path.pop_back();
            sum -= i;
        }
    }
    
    vector<vector<int>> combinationSum3(int k, int n) {
        backtracking(k, n, 0, 1);
        return res;
    }
};

剪枝：

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

        for (int i = startIndex; i <= 9 - (k - path.size()) + 1; i++){
            sum += i;
            path.push_back(i);
            backtracking(k, n, sum, i + 1);
            path.pop_back();
            sum -= i;
        }
    }
    
    vector<vector<int>> combinationSum3(int k, int n) {
        backtracking(k, n, 0, 1);
        return res;
    }
};

Leetcode17.电话号码的字母组合

文章链接：代码随想录
题目链接：17.电话号码的字母组合

思路：定义一个数字与字母对应的数组或map，逐层递归回溯。

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

    vector<string> res;
    string s;
    void backtracking(const string& digits, int index){
        if (index == digits.size()){
            res.push_back(s);
            return ;
        }

        int digit = digits[index] - '0';
        string letter = lettermap[digit];
        for (int i = 0; i < letter.size(); i++){
            s.push_back(letter[i]);
            backtracking(digits, index + 1);
            s.pop_back();
        }
    }
    
    vector<string> letterCombinations(string digits) {
        if (digits.size() == 0) return res;
        backtracking(digits, 0);
        return res;
    }
};

或者可以把回溯过程直接放到递归函数里（照搬Carl哥代码了）

class Solution {
private:
        const string letterMap[10] = {
            "", // 0
            "", // 1
            "abc", // 2
            "def", // 3
            "ghi", // 4
            "jkl", // 5
            "mno", // 6
            "pqrs", // 7
            "tuv", // 8
            "wxyz", // 9
        };
public:
    vector<string> result;
    void getCombinations(const string& digits, int index, const string& s) { // 注意参数的不同
        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++) {
            getCombinations(digits, index + 1, s + letters[i]);  // 注意这里的不同
        }
    }
    vector<string> letterCombinations(string digits) {
        result.clear();
        if (digits.size() == 0) {
            return result;
        }
        getCombinations(digits, 0, "");
        return result;

    }
};

注意这里

void getCombinations(const string& digits, int index, const string& s){
...
getCombinations(digits, index + 1, s + letters[i]);
...
}

s传引用时，每层传出去的是s + letters[i]，也就是新的s接收的是原s的地址空间加上一段letter[i]， 所以原来的每层s都不会发生改变，保证了回溯的顺利进行。
若用值传递也是可以的，就是相当于每层的s都新开了一段地址空间，占用内存变多了，但每层的s也不会发生改变。
第二十五天打卡，两道题分两天做的，昨天拍冰层照片+课题组三国杀，C++的小项目得抓紧做了，加油！！！