凌云的博客

行胜于言

LeetCode 算法题 40. 组合总和 II

分类:algorithm| 发布时间:2016-09-30 18:03:00


题目

给定一个数组 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]
]

解法

解法跟上一道题很像,只需要做细微的修改。

class Solution {
public:
    vector<vector<int>> combinationSum2(vector<int>& candidates, int target) {
        sort(candidates.begin(), candidates.end());
        dfs(candidates, 0, target);
        return ret_;
    }

private:
    void dfs(vector<int> &candidates, int idx, int target) {
        if (target < 0) return;
        if (target == 0) {
            ret_.push_back(part_);
            return;
        }

        const int size = candidates.size();
        for (int i = idx; i < size; ++i) {
            if (i > idx && candidates[i] == candidates[i - 1]) continue;
            part_.push_back(candidates[i]);
            dfs(candidates, i + 1, target - candidates[i]);
            part_.pop_back();
        }
    }

    vector<int> part_;
    vector<vector<int>> ret_;
};