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

216.组合总和III

找出所有相加之和为 n 的 k 个数的组合，且满足下列条件：

• 只使用数字1到9

• 每个数字 最多使用一次

返回 所有可能的有效组合的列表 。该列表不能包含相同的组合两次，组合可以以任何顺序返回。

示例 1:

输入: k = 3, n = 7
输出: [[1,2,4]]
解释:
1 + 2 + 4 = 7
没有其他符合的组合了。

提示:

• 2 <= k <= 9

• 1 <= n <= 60

思路

与上一题思路一致，多一个 sum 记录 path 组合内总和

剪枝操作主要是如果 path 里总和已经大于 n 了，就剪掉

代码实现

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

17.电话号码的字母组合

给定一个仅包含数字 2-9 的字符串，返回所有它能表示的字母组合。答案可以按 任意顺序 返回。

给出数字到字母的映射如下（与电话按键相同）。注意 1 不对应任何字母。

示例 1：

输入：digits = "23"
输出：["ad","ae","af","bd","be","bf","cd","ce","cf"]

提示：

• 0 <= digits.length <= 4

• digits[i] 是范围 ['2', '9'] 的一个数字。

思路

先把数字对应的字母映射存一下

回溯的思路也和前两题类似，要注意的事 index 的部分和之前的 startIndex 不一样，而是代表了输入的是第几个数字

终止条件也就是所有的遍历完毕

剪枝操作是在函数调用的时候，判断 digits 是否为空

代码实现

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;
    string s;
    void backtracking(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) {
        if(digits.size() == 0){
            return result;
        }
        backtracking(digits, 0);
        return result;
    }
};