Day25 回溯 LeetCode 216 17

216. 组合总和 III

心得

• 回溯的终止条件单独处理，勿在单层处理逻辑那边绕

题解

• 回溯三部曲，在组合基础上对终止条件新增求和判断

class Solution {
public:
    vector<vector<int>> result;
    vector<int> path;
    void backtracing(int targetSum, int k, int sum, int startIndex) {
        if (sum > targetSum) return; // 剪枝
        if (path.size() == k ) {
            if (targetSum == sum) {
                result.push_back(path); 
            }
            return;
        }
				// 剩余长度剪枝
        for (int i = startIndex; i <= 9 - (k - path.size()) + 1; i++) {
            sum += i;
            path.push_back(i);
            backtracing(targetSum, k, sum, i + 1);
            sum -= i; // 回溯
            path.pop_back(); // 回溯
        }
    }
    vector<vector<int>> combinationSum3(int k, int n) {
        path.clear();
        result.clear();
        backtracing(n, k, 0, 1);
        return result;
    }
};

17. 电话号码的字母组合

心得

• 多循环问题看到还是考虑到回溯，但是参数没考虑好，最好画图示意

题解

• 注意数字字母映射，终止条件的判断，已经遍历下一个位置的考虑
• 本题区别余前两天在于为不同集合，而不是相同集合，所以没有通过全局startIndex来控制选择的区域，而是通过index来负责取不同集合中的元素

class Solution {
private:
    const string letterMap[10] = {
        "",
        "",
        "abc",
        "def",
        "ghi",
        "jkl",
        "mno",
        "pqrs",
        "tuv",
        "wxyz",
    };
public:
    string path;
    vector<string> result;
    void backtracing(const string& digits, int index) {
        if (index == digits.size()) { // 跟size对比，保证最后一个正确的字母已经被上次记录
            result.push_back(path); 
            return;
        }
        int digit = digits[index] - '0'; // 取对应数字的int
        string letters = letterMap[digit];
        for (int i = 0; i < letters.size(); i++) {
            path.push_back(letters[i]);
            backtracing(digits, index + 1); // 遍历下一个字母
            path.pop_back(); // 回溯
        }
    }
    vector<string> letterCombinations(string digits) {
        path.clear();
        result.clear();
        if (digits.size() == 0) return result;
        backtracing(digits, 0);
        return result;
    }
};