代码随想录Day25 | 216.组合总和III、17.电话号码的字母组合

216.组合总和III

深度由k控制

宽度由数组中的元素个数控制

注意本题中有两个剪枝操作,还有回溯的过程也不要忘记

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

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

17.电话号码的字母组合

树的深度由输入数字的个数决定

有一个隐藏回溯需要注意

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

你可能感兴趣的:(c++,算法,数据结构)