LeetCode 与组合数相关的题目

最新推荐文章于 2024-10-17 19:06:39 发布

Hanknet

最新推荐文章于 2024-10-17 19:06:39 发布

阅读量108

点赞数

文章标签： leetcode 算法 c++

本文链接：https://blog.csdn.net/Hanknet/article/details/130449054

版权

216. 组合总和 III

方法：递归

class Solution {
private:
    vector<vector<int>> res;
    vector<int> path;
    void solve(int k, int goal, int cur, int idx) {
        if (cur > goal) return;
        if (path.size() == k) {
            if (cur == goal) res.push_back(path);
            return ;
        }

        for (int i = idx; i <= 9-(k-path.size()) + 1; ++i) {
            path.push_back(i);
            solve(k, goal, cur + i, i + 1);
            path.pop_back();
        }
    }
public:
    vector<vector<int>> combinationSum3(int k, int n) {
        res.clear();
        path.clear();
        solve(k, n, 0, 1);
        return res;
    }
};

$时间复杂度O( $\binom{n}{k}$ *k),空间复杂度(n)

17. 电话号码的字母组合

方法：递归

class Solution {
private:
    const string letterMap[10] = {
        "",
        "",
        "abc",
        "def",
        "ghi",
        "jkl",
        "mno",
        "pqrs",
        "tuv",
        "wxyz",
    };
    vector<string> res;
    string str;
    void solve(const string& digits, int idx, const string& str) {
        if (idx == digits.size()) {
            res.push_back(str);
            return ;
        }
        int num = digits[idx] - '0';
        string word = letterMap[num];
        for (int i = 0; i < word.size(); ++i) {
            solve(digits, idx + 1, str+word[i]);
        }
    }
public:
    vector<string> letterCombinations(string digits) {
        int n = digits.size();
        if (!n) return res;
        solve(digits, 0, str);
        return res;
    }
};

$时间复杂度O( $3^{6}*4^{^{2}}$ ),空间复杂度(n)