LeetCode 40. 组合总和 II（Combination Sum II）

题目描述

 

给定一个数组 candidates 和一个目标数 target ，找出 candidates 中所有可以使数字和为 target 的组合。

candidates 中的每个数字在每个组合中只能使用一次。

说明：

• 所有数字（包括目标数）都是正整数。
• 解集不能包含重复的组合。 

示例 1:

输入: candidates = [10,1,2,7,6,1,5], target = 8,
所求解集为:
[
  [1, 7],
  [1, 2, 5],
  [2, 6],
  [1, 1, 6]
]

示例 2:

输入: candidates = [2,5,2,1,2], target = 5,
所求解集为:
[
  [1,2,2],
  [5]
]

 

解题思路

 

利用回溯思想，首先将数组从小到大排序，然后从第一个数开始，若当前数字和小于目标，则从当前索引的数开始向后依次加入到当前vector中，并更新当前索引和数字和，再从下一个索引重复此动作，直到遇到数字和等于目标。

 

代码

 

class Solution {
public:
    vector<vector<int>> combinationSum2(vector<int>& candidates, int target) {
        vector<vector<int>> res;
        vector<int> v;
        sort(candidates.begin(), candidates.end());
        cmb(candidates, target, v, 0, 0, res);
        return res;
    }
    void cmb(vector<int> candidates, int target, vector<int> &v, int idx, int sum, vector<vector<int>> &res){
        if(sum == target)
            res.push_back(v);
        else if(sum < target){
            for(int i = idx; i < candidates.size(); i++){
                if(i > idx && candidates[i] == candidates[i - 1])
                    continue;
                v.push_back(candidates[i]);
                cmb(candidates, target, v, i + 1, sum + candidates[i], res);
                v.pop_back();
            }
        }
    }
};