Leetcode 90. 子集 II C++回溯法

题目描述

给定一个可能包含重复元素的整数数组 nums，返回该数组所有可能的子集（幂集）。
说明：解集不能包含重复的子集。
示例:

输入: [1,2,2]
输出:
[
  [2],
  [1],
  [1,2,2],
  [2,2],
  [1,2],
  []
]

解答

使用回溯法，为了去掉重复需要先排序，并且使用set容器。

class Solution {
public:
    vector<vector<int>> subsetsWithDup(vector<int>& nums) {
        sort(nums.begin(),nums.end());
        set<vector<int>> res;
        vector<int> temp;
        fun(nums,0,temp,res);
        vector<vector<int>> result;
        for(auto mem:res)
        {
            result.push_back(mem);
        }
        return result;
    }
    void fun(const vector<int> &nums, int index,vector<int> &temp,set<vector<int>> &res)
    {
        res.insert(temp);
        while(index<nums.size())
        {
            temp.push_back(nums[index]);
            fun(nums,index+1,temp,res);
            temp.pop_back();
            ++index;
        }
    }
};