leetcode热题100（39. 组合总和）dfs剪枝c++

❦丿多像灬笑话、℡2025-01-03 13:50

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

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

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

示例 1：

复制代码

输入：candidates = [2,3,6,7], target = 7
输出：[[2,2,3],[7]]
解释：
2 和 3 可以形成一组候选，2 + 2 + 3 = 7 。注意 2 可以使用多次。
7 也是一个候选， 7 = 7 。
仅有这两种组合。

示例 2：

复制代码

输入: candidates = [2,3,5]
target = 8
输出: [[2,2,2,2],[2,3,3],[3,5]]

示例 3：

复制代码

输入: candidates = [2], target = 1
输出: []

题意

求candidates 数组的加起来的数等于target的所有情况，其中 candidates 中的数可以重复

思路

因为我们可以重复选同一个数，那么我们就可以对当前这个位置的选还是不选进行判断，数据最大为 taget<=40,那么其中最大只可能选重复40个位置的数，我们可以对candidates数组进行排序，那么大的数肯定是最小的选择了，我们就可以提前剪枝优化了

代码

cpp 复制代码

class Solution {
public:
    vector<vector<int>> ans;
    int n;
    vector<int> t;
    vector<vector<int>> combinationSum(vector<int>& a, int target) {
        n = a.size();
        t = a;
        sort(a.begin(),a.end());
        vector<int> tmp;
        dfs(tmp,0,0,target);
        return ans;
    }

    void dfs(vector<int> res,int st,int tmp,int target){
        //cout<<target<<" "<<tmp<<endl;
        if(st>=n || tmp>target) return;
        if(tmp==target){
            ans.push_back(res);
            return;
        }
        res.push_back(t[st]);
        dfs(res,st,tmp+t[st],target);
        res.pop_back();
        dfs(res,st+1,tmp,target);
        
    }

};