LeetCode 90. Subsets II

来源:互联网 发布:mac表格使用教程 编辑:程序博客网 时间:2024/06/06 01:51

LeetCode 90. Subsets II

原题目:
Given a collection of integers that might contain duplicates, nums, return all possible subsets.

Note: The solution set must not contain duplicate subsets.

For example,
If nums = [1,2,2], a solution is:

[  [2],  [1],  [1,2,2],  [2,2],  [1,2],  []]

题目翻译:
给出一个可能包含重复元素的整型数组,找到所有该数组的子集,返回的结果不能包含重复的子集。

思路:
这题是 LeetCode 78. Subsets的一道变式题,在原题的基础上增加了可能包含重复元素的限制。关于原题的详细解答可以参照本人另一篇文章:LeetCode 78. Subsets的三种解法。
这题同样也可以使用回溯法(Backtracking)和迭代法(Iterative)解决,而位操作(Bit Manipulation)需要事先知道返回结果集的大小,除非先进行一次遍历求得结果集大小,而且各个变量的中间值也不是2的幂次方,因此就不在此给出位操作的方案。

要点:
1)尽量从原题的基础上进行思考,找出其中的区别,在原代码上修改,利用题目和题目之间的关联性。
2)这道题需要先对原数组进行排序,目前还不知道不排序该怎么做。

方案一:回溯(Backtracking)
1)可以在递归之前用一个unordered_set保存出现过的元素。
AC代码如下:(C++, 13ms)

class Solution {public:    void backtrack(int start, vector<int>& sub, vector<int>& nums, vector<vector<int>>& subs) {        subs.push_back(sub);        unordered_set<int> uset;        for (int i = start; i < nums.size(); i++) {            if (uset.find(nums[i]) == uset.end()) {                sub.push_back(nums[i]);                backtrack(i + 1, sub, nums, subs);                sub.pop_back();                uset.insert(nums[i]);            }        }    }    vector<vector<int>> subsetsWithDup(vector<int>& nums) {        vector<vector<int>> subs;        vector<int> sub;        sort(nums.begin(), nums.end());        backtrack(0, sub, nums, subs);        return subs;    }};

2)由于数组事先已经过排序,因此不需要再用额外的unordered_set去判断重复元素,直接判断nums[i]和nums[i - 1]是否相等就行了。
AC代码如下:(C++, 13ms)

class Solution {public:    void backtrack(int start, vector<int>& sub, vector<int>& nums, vector<vector<int>>& subs) {        subs.push_back(sub);        for (int i = start; i < nums.size(); i++) {            if (i == start || nums[i] != nums[i - 1]) {                sub.push_back(nums[i]);                backtrack(i + 1, sub, nums, subs);                sub.pop_back();            }        }    }    vector<vector<int>> subsetsWithDup(vector<int>& nums) {        vector<vector<int>> subs;        vector<int> sub;        sort(nums.begin(), nums.end());        backtrack(0, sub, nums, subs);        return subs;    }};

方案二:迭代(Iterative)
仍旧参照LeetCode 78. Subsets的三种解法中方案二的思想,判断下一个元素是否是重复元素:
若nums[i - 1] != nums[i],则直接按照方案二来做;
若nums[i - 1] == nums[i],则只复制最后增加的那部分子集,即将内循环迭代的初始值移动至最后一部分。
AC代码如下:(C++, 6ms)

class Solution {public:    vector<vector<int> > subsetsWithDup(vector<int> &nums) {        sort(nums.begin(), nums.end());        vector<vector<int>> subs = {{}};        for (int i = 0; i < nums.size(); i++) {            int start, size;            if (i == 0 || nums[i] != nums[i - 1])   start = 0;            else                                    start = size;            size = subs.size();            for (int j = start; j < size; j++) {                subs.push_back(subs[j]);                subs.back().push_back(nums[i]);            }        }        return subs;    }};
0 0
原创粉丝点击