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 I题目的基础上增加了重复的数字,相应的,我们在Subsets I基础上去重就可以了。
因为每种集合都是排过序的,因此对ret进行排序后相同的集合都处于相邻位置,然后去掉相邻的重复元素即可。
如果使用stl库的sort函数需要自己编写合适cmp函数。去重我用的是unique函数,unique函数将多余的重复元素放到数组后面,返回重复元素的起始位置的迭代器,所以还要使用erase将这些重复的元素删除掉。
AC code:
bool cmp(const vector<int> &vi1, const vector<int> &vi2)
{
int len1 = vi1.size();
int len2 = vi2.size();
int i, j;
i = j = 0;
while (i < len1&&j < len2)
{
if (vi1[i] != vi2[j])
return vi1[i] < vi2[j];
else
{
i++;
j++;
}
}
if (len1 == len2)
return false;
if (i == len1)
return true;
return false;
}
class Solution {
public:
vector<vector<int> > subsetsWithDup(vector<int> &S)
{
vector<vector<int>> ret;
int n = S.size();
if (n == 0)
return ret;
int int_size = sizeof(int)* 8;
int *flag = new int[(n + 1) / int_size + 1]; //多分配了一个1bit大小,方便做结束判断
memset(flag, 0, ((n + 1) / int_size + 1)*sizeof(int));
vector<int> temp;
while ((flag[(n + 1) / int_size] & (1 << n % int_size)) == 0)
{
temp.clear();
for (int i = 0; i < n; i++)
{
if ((flag[i / int_size] & (1 << i%int_size)) != 0)
temp.push_back(S[i]);
}
sort(temp.begin(), temp.end());
ret.push_back(temp);
int j = 0;
while (1) //加1操作
{
if (flag[j] != INT_MAX)
{
++flag[j];
break;
}
else
flag[j++] = 0;
}
}
sort(ret.begin(), ret.end(), cmp);
auto iter = unique(ret.begin(), ret.end());
ret.erase(iter, ret.end());
return ret;
}
};