LeetCode数组篇39之回溯法

 39. 组合总和

难度中等

给你一个 无重复元素 的整数数组 candidates 和一个目标整数 target ，找出 candidates 中可以使数字和为目标数 target 的 所有 不同组合 ，并以列表形式返回。你可以按 任意顺序 返回这些组合。

candidates 中的 同一个 数字可以 无限制重复被选取 。如果至少一个数字的被选数量不同，则两种组合是不同的。 

对于给定的输入，保证和为 target 的不同组合数少于 150 个。

破天荒地代码一次通过~，不过效率TCL

class Solution {
public:
    vector<vector<int>> res;
    void util(vector<int>& candidates,int i,int n,int sum,int target,vector<int> temp){
        if(i==n || sum>target)
            return;
        if(sum==target){
            res.push_back(temp);
            return;
        }
        temp.push_back(candidates[i]);
        util(candidates,i,n,sum+candidates[i],target,temp);
        temp.pop_back();
        util(candidates,i+1,n,sum,target,temp);
    }
    vector<vector<int>> combinationSum(vector<int>& candidates, int target) {
        sort(candidates.begin(),candidates.end());
        int n=candidates.size();
        vector<int> temp;
        util(candidates,0,n,0,target,temp);
        return res;
    }
};

 用的回溯法，怎么耗时这么长哇，有简单的剪枝，emmm，再瞅瞅

永远相信官方解答下面评论区的大佬，参考其中一个评论，对代码进行优化，其实上一段代码之所以慢，是因为其对数字的选择和每个数字个数的试探都采用了dfs，而dfs实现方式是栈，数据多时，这个栈被撑得很大很大，不能及时回收，优化思路是，对数字种类的试探采用for循环，对数字个数的试探采用回溯，代码如下啦~

class Solution {
public:
    vector<vector<int>> res;
    void util(vector<int>& candidates,int begin,int n,int sum,int target,vector<int> temp){
        if(sum>target)
            return;
        if(sum==target){
            res.push_back(temp);
            return;
        }
        for(int i=begin;i<n;i++){
            if(sum+candidates[i]>target)
                return;
            temp.push_back(candidates[i]);
            util(candidates,i,n,sum+candidates[i],target,temp);
            temp.pop_back();
        }
    }
    vector<vector<int>> combinationSum(vector<int>& candidates, int target) {
        sort(candidates.begin(),candidates.end());
        int n=candidates.size();
        vector<int> temp;
        util(candidates,0,n,0,target,temp);
        return res;
    }
};

看看结果

 又刷了一题哎，又进步了哎，下一题见！😉👧