首页 > 代码库 > leetcode第一刷_Subsets II

leetcode第一刷_Subsets II

要求子集,有非常现成的方法。N个数,子集的个数是2^N,每个元素都有在集合中和不在集合中两种状态,这些状态用[0,pow(2,N)]中每个数来穷举,如果这个数中的第i位为1,说明当前集合中包含源数组中的第i个数。

至于有没有重复的元素,大部分有重复元素的问题,都可以借助一个vis集合,里面存放所有已经求得的集合或者其他形式的解,只有少数题目会超时,哪些问题具体的说。

class Solution {
public:
    vector<vector<int> > subsetsWithDup(vector<int> &S) {
        int msize = S.size();
        int mindex = pow(2, msize);
        vector<vector<int> > res;
        set<vector<int> > mset;
        vector<int> tpres;
        sort(S.begin(), S.end());
        for(int i=0;i<mindex;i++){
            int mask = 1;
            for(int pos=0;pos<msize;pos++){
                if(mask&i)  tpres.push_back(S[pos]);
                mask<<=1;
            }
            if(find(mset.begin(), mset.end(), tpres) == mset.end()){
                mset.insert(tpres);
                res.push_back(tpres);
            }
            tpres.clear();
        }
        return res;
    }
};