代码随想录算法训练营第25天 [216.组合总和III 17.电话号码的字母组合]

代码随想录算法训练营第25天 [216.组合总和III 17.电话号码的字母组合]

---

一、216.组合总和III

链接: 代码随想录.
思路：回溯剪枝
做题状态：看解析后做出来了

class Solution
{
public:
    vector<int> path;
    vector<vector<int>> result;
    vector<vector<int>> combinationSum3(int k, int n)
    {
        if (k > n)
        {
            return result;
        }
        backtracking(k, n, 1);
        return result;
    }

    void backtracking(int k, int n, int start_index)
    {
        int sum = 0;
        for (auto i : path)
        {
            sum += i;
        }
        if (sum > n)
        {
            return;
        }
        if (path.size() == k)
        {
            if (sum == n)
            {
                result.push_back(path);
            }
            return;
        }
        /*剪枝操作
         所有元素共有1-9        9
         总共需要的元素个数     k
         我已经选择到了         i
         我已经选择的元素个数   path.size()
         我还需要多少个元素     k-path.size()
         待选择列表中剩下的元素  9-(k-path.size())


         */
        for (int i = start_index; i <= 9 - (k - path.size()) + 1; i++)
        {
            path.push_back(i);
            backtracking(k, n, i + 1);
            path.pop_back();
        }
    }
};

二、17.电话号码的字母组合

链接: 代码随想录.
思路：回溯
做题状态：看解析后做出来了

class Solution
{
public:
    string mp[10] = {"", "", "abc", "def", "ghi", "jkl", "mno", "pqrs", "tuv", "wxyz"};
    vector<string> result;
    string path;
    vector<string> letterCombinations(string digits)
    {
        if (digits.size() == 0)
        {
            return result;
        }
        int index = 0;
        backtracking(index, digits);
        return result;
    }

    void backtracking(int index, string digits)
    {
        if (path.size() == digits.size())
        {
            result.push_back(path);
            return;
        }
        int digit = digits[index] - '0';
        string letters = mp[digit];
        for (int i = 0; i < letters.size(); i++)
        {
            path.push_back(letters[i]);
            backtracking(index + 1, digits);
            path.pop_back();
        }
    }
};