凌云的博客

行胜于言

LeetCode 算法题 39. 组合总和

分类:algorithm| 发布时间:2016-09-30 15:27:00


题目

给定一个 无重复元素 的数组 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]
]

解法

由于题目保证了输入的数组是无重复,可以简单地使用 dfs 解决这个问题。

class Solution {
public:
    vector<vector<int>> combinationSum(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) {
            part_.push_back(candidates[i]);
            dfs(candidates, i, target - candidates[i]);
            part_.pop_back();
        }
    }

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