90.子集II

2018-11-22  本文已影响0人  HITZGD

题目
给定一个可能包含重复元素的整数数组 nums,返回该数组所有可能的子集(幂集)。

说明:解集不能包含重复的子集。

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

思路
本体沿用78.子集的代码,不同之处为加了判断重复项的代码,如下:

            vector<vector<int>>::iterator flag;
            flag = find(result.begin(), result.end(), temp);
            if (flag == result.end())
            {
                result.push_back(temp);
                return;
            }

完整代码为:

#include <vector>
#include <algorithm>
using namespace std;
class Solution {
public:
    vector<vector<int>> subsetsWithDup(vector<int>& nums) {
        vector<vector<int>> result;
        vector<int>temp;
        int size = nums.size();
        int start = 0;
        sort(nums.begin(), nums.end());
        for (int i = 0; i <= size; i++)
        {
            combine(start, i, result, temp, nums);
        }
        return result;
    }
    void combine(int start, int size, vector<vector<int>>& result, vector<int>& temp, vector<int> nums)
    {
        if (temp.size() == size)
        {
            sort(temp.begin(), temp.end());
            vector<vector<int>>::iterator flag;
            flag = find(result.begin(), result.end(), temp);
            if (flag == result.end())
            {
                result.push_back(temp);
                return;
            }
        }
        for (int i = start; i < nums.size(); i++)
        {
            temp.push_back(nums[i]);
            combine(i + 1, size, result, temp, nums);
            temp.pop_back();
        }
    }
};

int main(int argc, char* argv[])
{
    vector<int>nums = { 0,1,4 };
    auto res = Solution().subsetsWithDup(nums);
    return 0;
}

上一篇 下一篇

猜你喜欢

热点阅读