leetcode 刷题之路 57 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 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;
	}
	
};