Leetcode第40题(组合总和2)

题目描述:

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

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

注意：解集不能包含重复的组合。

示例 1:

输入: candidates = [10,1,2,7,6,1,5], target = 8,
输出:[[1,1,6],[1,2,5],[1,7],[2,6]]
示例 2:

输入: candidates = [2,5,2,1,2], target = 5,
输出:[[1,2,2],[5]]

 

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

    void dfs(vector<int>& candidates,int target,int idx){
        //target == 0
        if(target == 0){
            ans.push_back(combine);
            return;
        }
        //idx == candidates.size()
        if(idx == candidates.size()) return;

        //统计相同元素的个数
        int cnt = 1;
        while(idx + 1 < candidates.size() && candidates[idx] == candidates[idx + 1]){
            cnt++;
            idx++;
        }
        //避免重复
        for(int i = 0;i <= cnt && target >= i * candidates[idx];i++){
            dfs(candidates,target - i * candidates[idx],idx + 1);
            combine.push_back(candidates[idx]);
        }
        for(int i = 0;i <= cnt && target >= i * candidates[idx];i++)
            combine.pop_back();
    }
};