1. 程式人生 > >[leetcode]Subsets II

[leetcode]Subsets II

size solution set ron style 方法 lec [1] elements

問題描寫敘述:

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