代码随想录训练营第22天|回溯剪枝

最新推荐文章于 2024-10-09 09:53:23 发布

jiyisuifeng1991

最新推荐文章于 2024-10-09 09:53:23 发布

阅读量203

点赞数 2

文章标签：深度优先算法图论

本文链接：https://blog.csdn.net/jiyisuifeng1991/article/details/141882606

版权

77. 组合

class Solution {

public:
    vector<vector<int>> result;
    vector<int> path;

    void dfs(int n, int k, int startIdx){
        if(path.size()==k){
            result.push_back(path);
            return;
        }
        for(int i=startIdx; i<=n+1-(k-path.size()); i++){
            path.push_back(i);
            dfs(n,k,i+1);
            path.pop_back();
        }
    }
    vector<vector<int>> combine(int n, int k) {
        dfs(n, k, 1);
        return result;
    }
};

减枝策略：i不能太大，否则无法凑够k个元素，当前已收集元素个数:path.size()，待收集个数：k-path.size()，终点最大为(开区间): n+1，得到起点最大：n+1-(k-path.size())。

216. 组合总和 III

class Solution {
public:
    vector<vector<int>> result;
    vector<int> path;
    int sum;

    void dfs(int k, int n, int startIdx){
        if(path.size()==k){
            if(sum==n){
                result.push_back(path);
            }
            return;
        }
        for(int i=startIdx; i<=10-(k-path.size())&&sum+i<=n; i++){
            path.push_back(i);
            sum+=i;
            dfs(k,n,i+1);
            sum-=i;
            path.pop_back();
        }
    }
    vector<vector<int>> combinationSum3(int k, int n) {
        dfs(k,n,1);
        return result;
    }
};

剪枝策略：i不能太大，考虑两个限制：1.满足组合个数要求，如果i太大，无法凑到k个元素 2.满足求和要求，如果i太大，当前层已经突破了n，因为i是递增的(i++)，所以后续的遍历都是无效的。

17. 电话号码的字母组合

class Solution {
public:
    unordered_map<char,string> mp{{'2',"abc"},{'3',"def"},{'4',"ghi"},{'5',"jkl"},{'6',"mno"},{'7',"pqrs"},{'8',"tuv"},{'9',"wxyz"}};
    vector<string> result;
    string path;

    void dfs(string& digits, int startIdx){
        if(path.size()==digits.size()){
            result.push_back(path);
            return;
        }
        for(auto& c: mp[digits[startIdx]]){
            path.push_back(c);
            dfs(digits,startIdx+1);
            path.pop_back();
        }
    }
    vector<string> letterCombinations(string digits) {
        if(digits.empty())
            return {};
        dfs(digits,0);
        return result;
    }
};

startIdx标记处理到digits哪一位，通过mp查询当前字符digits[startIdx]被翻译的候选集，for循环保证当前层所有候选字符可以被选到，递归dfs保证所有digits依次被处理完。