39. 组合总和 – 力扣(LeetCode)

题目描述

给定一个无重复元素的数组 candidates 和一个目标数 target ,找出 candidates 中所有可以使数字和为 target 的组合。
candidates 中的数字可以无限制重复被选取。
说明:
所有数字(包括 target)都是正整数。
解集不能包含重复的组合。
示例 1:
输入: candidates = [2,3,6,7], target = 7,
所求解集为:
[
[7],
[2,2,3]
]
示例 2:
输入: candidates = [2,3,5], target = 8,
所求解集为:
[
  [2,2,2,2],
  [2,3,3],
  [3,5]
]

题解

递归搜索+剪枝
详细见下图,图是以示例1为例的。

代码

class Solution {
public:
    //递归+剪枝
    vector<vector<int>>res;
    vector<int>sol;
    void dfs(vector<int>& nums, int beg, int target){
        for(int i = beg; i < nums.size(); ++i){
            sol.push_back(nums[i]);
            if(target - nums[i] > 0){//继续向下搜索
                dfs(nums, i, target - nums[i]);
            }
            else if(target - nums[i] == 0){//将符合题意的解放到res中
                res.push_back(sol);
            }
            sol.pop_back();//回溯的时候记得弹出最后一个元素
            if(target - nums[i] < 0){//剪枝,如果当前的sol中所有元素之和大于target,则直接return
                return;
            }
        }
    }
    vector<vector<int>> combinationSum(vector<int>& candidates, int target) {
        sort(candidates.begin(), candidates.end());
        dfs(candidates, 0, target);
        return res;
    }
};

执行结果

转载请注明:Onwaier‘s Blog » 39. 组合总和 – 力扣(LeetCode)