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

最新推荐文章于 2024-11-11 22:10:43 发布

WynnLu

最新推荐文章于 2024-11-11 22:10:43 发布

阅读量43

点赞数

文章标签：算法深度优先

本文链接：https://blog.csdn.net/WynnLu/article/details/133914006

版权

文章讲述了如何使用回溯算法（DFS）解决LeetCode中的两道题目：组合总和III（找出k个数之和为n的组合）和电话号码的字母组合（将数字字符串转换为字母组合）。作者提供了C++代码实现，展示了如何构建四叉树来处理不满4个的数字字符。

摘要由CSDN通过智能技术生成

216.组合总和III

题目描述

题目链接：力扣216.组合总和III

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

只使用数字1到9
每个数字最多使用一次
返回所有可能的有效组合的列表。该列表不能包含相同的组合两次，组合可以以任何顺序返回。

思路

回溯，从9个数中选择k个数，将其中满足和为n的结果记录下来。

代码实现

class Solution {
public:
    vector<int> temp;
    vector<vector<int>> ans;

    void dfs(int cur, int n, int k, int sum) {
        if (temp.size() + (n - cur + 1) < k || temp.size() > k) {
            return;
        }
        if (temp.size() == k && accumulate(temp.begin(), temp.end(), 0) == sum) {
            ans.push_back(temp);
            return;
        }
        temp.push_back(cur);
        dfs(cur + 1, n, k, sum);
        temp.pop_back();
        dfs(cur + 1, n, k, sum);
    }

    vector<vector<int>> combinationSum3(int k, int n) {
        dfs(1, 9, k, n);
        return ans;
    }
};

17.电话号码的字母组合

题目描述

题目链接：力扣17.电话号码的字母组合

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

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

思路

做一棵四叉树，每个数字代表4个字母，不满4个的填充一个无意义字符，如此每个叶子节点代表一个字母组合。

代码实现

class Solution {
public:
    char digit2alpha[8][4] = {
        {'a', 'b', 'c', '-'},
        {'d', 'e', 'f', '-'},
        {'g', 'h', 'i', '-'},
        {'j', 'k', 'l', '-'},
        {'m', 'n', 'o', '-'},
        {'p', 'q', 'r', 's'},
        {'t', 'u', 'v', '-'},
        {'w', 'x', 'y', 'z'}
    };
    vector<string> res;
    void dfs(string s, string digits, int index){
        if(index == digits.size()){
            res.push_back(s);
            return ;
        }
        int num = digits[index] - '0' - 2;
        for(int i=0; i<4; i++){
            if(digit2alpha[num][i] != '-'){
                dfs(s+digit2alpha[num][i], digits, index+1);
            }
        }
    }
    vector<string> letterCombinations(string digits) {
        if(digits.size() == 0){
            vector<string> null;
            return null;
        }
        dfs("", digits, 0);
        return res;
    }
};