【刷题打卡】39. 组合总和

88 阅读1分钟

携手创作,共同成长!这是我参与「掘金日新计划 · 8 月更文挑战」的第27天,点击查看活动详情

组合总和

39. 组合总和 - 力扣(LeetCode) (leetcode-cn.com)

给你一个 无重复元素 的整数数组 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
输出: []

 

提示:

  • 1 <= candidates.length <= 30
  • 1 <= candidates[i] <= 200
  • candidate 中的每个元素都 互不相同
  • 1 <= target <= 500

解题思路

  1. 首先进行预处理,对原数组进行排序和删除比target大的数据
  2. 使用递归和回溯,将和为target的temp数组加入res;
  3. 剪枝:temp加入新数据的和大于target了;
  4. 去重:一开始会有重复出现,看其他答案受到启发,可以通过大小来判断,不让比之前小的数据加入temp;

代码实现

class Solution {
public:
    vector<vector<int>> combinationSum(vector<int>& candidates, int target) {
        sort(candidates.begin(),candidates.end());
        vector<int> candidates_new;
        vector<vector<int>> res;
        vector<int> temp;
        for(int i=0;i<candidates.size();i++){ // 删掉比target大的数据
            if(candidates[i]>target) break;
            candidates_new.push_back(candidates[i]);
        }
        dfs(candidates_new,res,temp,target);
        return res; 
    }
    void dfs(vector<int>& candidates_new,vector<vector<int>>& res,vector<int>& temp, int target){
        if(accumulate(temp.begin(),temp.end(),0)==target){
            res.push_back(temp);
        }
        for(int i=0;i<candidates_new.size();i++){
            if(temp.size()>0&&candidates_new[i]<temp[temp.size()-1]) continue; //一开始会有重复出现,看其他答案受到启发,可以通过大小来判断,不让比之前小的数据加入temp;
            temp.push_back(candidates_new[i]);
            if(accumulate(temp.begin(),temp.end(),0)>target){
                temp.pop_back();
                continue;
            }
            dfs(candidates_new,res,temp,target);
            temp.pop_back();
        }
    }
};