代码随想录算法训练营第二十二天|leetcode17、216题

leetcode第216题

class Solution {
private:
    vector<vector<int>> result; // 存放结果集
    vector<int> path; // 符合条件的结果

    // backtracking函数用于递归搜索符合条件的组合
    // targetSum：目标和，也就是题目中的n。
    // k：题目中要求k个数的集合。
    // sum：已经收集的元素的总和，也就是path里元素的总和。
    // startIndex：下一层for循环搜索的起始位置。
    void backtracking(int targetSum, int k, int sum, int startIndex) {
        // 当path的大小等于k时，表示已经找到了一个符合条件的组合
        if (path.size() == k) {
            // 如果该组合的元素总和等于目标和，则将其加入结果集中
            if (sum == targetSum)
                result.push_back(path);
            return; // 如果path.size() == k 但sum != targetSum 直接返回
        }
        
        // 从startIndex开始循环搜索符合条件的数字
        for (int i = startIndex; i <= 9; i++) {
            sum += i; // 将当前数字i加入到sum中
            path.push_back(i); // 将当前数字i加入到path中
            // 递归搜索下一层，startIndex从i+1开始，避免重复使用数字
            backtracking(targetSum, k, sum, i + 1); // 注意i+1调整startIndex
            sum -= i; // 回溯，将当前数字i从sum中移除
            path.pop_back(); // 回溯，将当前数字i从path中移除
        }
    }

public:
    // combinationSum3函数用于计算符合条件的组合
    vector<vector<int>> combinationSum3(int k, int n) {
        // 调用backtracking函数开始搜索符合条件的组合
        backtracking(n, k, 0, 1);
        // 返回结果集
        return result;
    }
};

leetcode第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<string> 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<string> letterCombinations(string digits) {
        s.clear();      // 清空临时字符串
        result.clear(); // 清空结果集

        // 如果输入的数字字符串为空，则直接返回空的结果集
        if (digits.size() == 0) {
            return result;
        }

        // 调用递归函数开始生成字母组合
        backtracking(digits, 0);

        // 返回最终的结果集
        return result;
    }
};