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

MoonBlue:)

已于 2023-05-19 21:03:29 修改

阅读量221

点赞数

分类专栏：代码随想录训练营文章标签：算法 c++ java

于 2023-05-19 17:26:31 首次发布

本文链接：https://blog.csdn.net/Hireath_/article/details/130770137

版权

代码随想录训练营专栏收录该内容

48 篇文章 1 订阅

订阅专栏

文章目录

216.组合总和III
17.电话号码的字母组合

216.组合总和III

思路

别忘了处理过程和回溯过程是一一对应的，处理有加，回溯就要有减！

代码

class Solution {
private:
    vector<vector<int>> result;
    vector<int> path;
    void backtracing(int k, int n, int startidx, int sum) {
        if(path.size() == k) {
            if(sum == n) result.push_back(path);
                return;
        }

        for(int i = startidx; i <= 9; i++) {
            sum += i;
            path.push_back(i);
            backtracing(k,n,i+1,sum);
            sum -= i;
            path.pop_back();
        }
        

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

剪枝后：

class Solution {
private:
    vector<vector<int>> result;
    vector<int> path;
    void backtracing(int k, int n, int startidx, int sum) {
        if(path.size() == k) {
            if(sum == n) result.push_back(path);
                return;
        }
        if(sum > n) {
            return;
        }

        for(int i = startidx; i <= 9; i++) {
            sum += i;
            path.push_back(i);
            backtracing(k,n,i+1,sum);
            sum -= i;
            path.pop_back();
        }
        

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

总结

画图后分析剪枝会更清晰

17.电话号码的字母组合

思路

代码

class Solution {
private:
    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();
        }
        return;
    }
public:
    const string letterMap[10] = {
        "", // 0
        "", // 1
        "abc", // 2
        "def", // 3
        "ghi", // 4
        "jkl", // 5
        "mno", // 6
        "pqrs", // 7
        "tuv", // 8
        "wxyz", // 9
    };
    vector<string> letterCombinations(string digits) {
        result.clear();
        s.clear();
        if(digits.size() == 0) {
            return result;
        }
        backtracking(digits,0);
        return result;
    }
};

总结

具体实现有点绕
注意这里for循环，可不像是在回溯算法：求组合问题和回溯算法：求组合总和中从startIndex开始遍历的。因为本题每一个数字代表的是不同集合，也就是求不同集合之间的组合，而77. 组合和216.组合总和III都是求同一个集合中的组合！每一层都是在一个新集合里搜索
注意：输入1 * #按键等等异常情况。代码中最好考虑这些异常情况，但题目的测试数据中应该没有异常情况的数据，所以我就没有加了。但是要知道会有这些异常，如果是现场面试中，一定要考虑到！