首页 > 代码库 > [leetcode]Subsets II

[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》的一个变形,区别在于上一篇是对set中的元素求子集,而本篇是对集合中的元素求子集。集合中的元素可能有重复的元素。利用上一篇的方法产生的子集可能还有重复的子集。只需在加入子集时查看是否已经包含了相同的子集即可。


代码:

vector<vector<int> > subsetsWithDup(vector<int> &S) {  c++
        vector<vector<int> > result;
        map<vector<int>,int > vecMap;
        int size = S.size();
        //add empty subset
        vector<int> tmp;
        result.push_back(tmp);
        
        int bound = 1<<size;
        for(int i = 1; i < bound; i++){
            vector<int> vec;
            int num = i;
            for(int j = size; j > 0 && num >0; j-- ){
                int tmp = (1 << j-1);
                if(num >= tmp){
                    vec.push_back(S[j-1]);
                    num -= tmp;
                }
                
            }
            sort(vec.begin(),vec.end());
            if(vecMap.find(vec)== vecMap.end()){
                result.push_back(vec);
                vecMap.insert(make_pair(vec,1));   
            }
        }
        return result;
    }


[leetcode]Subsets II