[Leetcode] subsets ii 求数组所有的子集

Given a collection of integers that might contain duplicates, S, return all possible subsets.

Note:

• Elements in a subset must be in non-descending order.
• The solution set must not contain duplicate subsets.

 

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

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

 题意：可能有重复的元素，返回所有子集。

思路：这题是subsets的扩展。大致的思路也是相同的，先是对给定整数进行排序，这样就可以保证所求的子集都是非降序的；在循环或递归的过程中跳过重复的。DFS法的代码如下：

class Solution {
public:
    vector<vector<int> > subsets(vector<int> &S) 
    {
        vector<vector<int>> res;
        vector<int> midArray;
        sort(S.begin(),S.end());
        getSubsets(S,0,midArray,res);

        return res;    
    }

    void getSubsets(vector<int> &S,int beg,vector<int> &midArray,vector<vector<int>> &res)
    {
        res.push_back(midArray);
        for(int i=beg;i<S.size();++i)
        {
            midArray.push_back(S[i]);
            getSubsets(S,i+1,midArray,res);
            midArray.pop_back();

            while(S[i]==S[i+1]) //跳过重复的元素
                i++;
        }
    }
};

 

迭代法：代码来源Grandyang的博客。其主要思路是，定义变量last记录下有重复的开始，因为第一个重复的元素，已经和之前res中的区间形成新的一部分，第二或其以后的重复元素，只要和新形成的部分结合就好，因为之前的已经结合过了。代码如下：

class Solution {
public:
    vector<vector<int> > subsets(vector<int> &S) 
    {
        vector<vector<int>> res(1);
        if(S.size()==0) return res;

        sort(S.begin(),S.end());
        int size=res.size();
        if(S.size()==0) return res;
        int last=S[0];

        for(int i=0;i<S.size();++i)
        {
            if(last !=S[i])
            {
                last=S[i];
                size=res.size();
            }

            int newSize=res.size();
            for(int j=newSize-size;j<newSize;++j)
            {
                res.push_back(res[j]);
                res.back().push_back(S[i]);
            }
        }
        return res;
    }
};