LeetCode 40 组合总和 C++ 回溯法

题目描述

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

思路

方法和leetcode39相同，Leetcode 39组合总和 C++ 回溯法。通过循环回溯。只不过本题所给元素中存在重复元素，那么就会导致得到的结果中存在重复。
主要是要去重

解答

class Solution {
public:
    vector<vector<int>> combinationSum2(vector<int>& candidates, int target) {
        set<vector<int>> res;
        vector<int> temp;
        sort(candidates.begin(),candidates.end());//排序是需要的
        fun(candidates,target,0,temp,res);
        vector<vector<int>> result;
        for(auto mem:res)
        {
            result.push_back(mem);
        }
        return result;
    }
    
    void fun(const vector<int> &candidates,int target,int index,vector<int> &temp,set<vector<int>> &res)
    {
        if(target<0) return;
        if(target==0)
        {
            res.insert(temp);
            return;
        }
        while(index<candidates.size())
        {
            temp.push_back(candidates[index]);
            fun(candidates,target-candidates[index],index+1,temp,res);
            temp.pop_back();
            ++index;
        }
    }
};