【Leetcode 90】C++ 4ms 回溯/位运算 难点除重的子集2

题目

Leetcode 90 子集2

解题思路

本题是子集1的变式，但其实套路和全排列问题相似。而难点在于除重！
解题思路都有两种：回溯和位运算。
解法一：含有标记数组的回溯（4ms,11.2MB）
解法二：不含标记，手动跳过的回溯（推荐！4ms,7.6MB）
解法三：位运算（推荐！4ms,7.6MB）

代码

解法一：含有标记数组的回溯（4ms,11.2MB）

先将nums数组进行排序，使相同的元素排在一起。
讨论的时候，先分析当前元素不在子集，再分析当前元素在子集中。
除重方法：引入vector flag数组
TIPS：
if (index >= 1 && nums[index] == nums[index - 1] && (temp.empty() || temp[temp.size() - 1] != nums[index]))这样不可以！因为会把[2,2,2]里面的[2,2]给排除掉，所以只能引入vector flag数组。

vector<vector<int>> result;
vector<bool> flag;//标记数组
void func(vector<int> &temp, int index, vector<int> &nums)
{
	if (index == nums.size())
		result.push_back(temp);
	else
	{
		func(temp, index + 1, nums);
		if(index >= 1 && nums[index] == nums[index - 1] && flag[index-1])
			return;//只能用return，因为反面太复杂！
		temp.push_back(nums[index]);
		flag[index] = false;
		func(temp, index + 1, nums);
		temp.pop_back();
		flag[index] = true;
	}
}
vector<vector<int>> subsetsWithDup(vector<int>& nums)
{
	sort(nums.begin(), nums.end());
	if (nums.size() == 0)
		return { {} };
	vector<int> temp;
	flag.assign(nums.size(), true);
	func(temp, 0, nums);
	return result;
}

解法二：不含标记，手动跳过的回溯（推荐！4ms,7.6MB）

先将nums数组进行排序，使相同的元素排在一起。
除重方法：当后一位元素与当前元素相同时，index++，跳过元素
TIPS：这个方法一定要先讨论元素在子集中，再讨论元素不在子集中。

vector<vector<int>> result;
void func(vector<int> &temp, int index, vector<int> &nums)
{
	if (index == nums.size())
		result.push_back(temp);
	else
	{
		temp.push_back(nums[index]);
		func(temp, index + 1, nums);
		temp.pop_back();
		while (index + 1 < nums.size() && nums[index] == nums[index + 1])//除重
			index++;
		func(temp, index + 1, nums);
	}
}
vector<vector<int>> subsetsWithDup(vector<int>& nums)
{
	sort(nums.begin(), nums.end());
	if (nums.size() == 0)
		return { {} };
	vector<int> temp;
	func(temp, 0, nums);
	return result;
}

解法三：位运算（推荐！4ms,7.6MB）

很容易想到：可以通过0,1表示是否取对应位置的元素。
除重方法：j != 0 && nums[j] == nums[j - 1] && ((i&(1 << (j - 1))) == 0)表示当前位置元素和前一位置元素相同，且前一位置元素并不在子集内时，需要除重！（分析前一元素的时候，已经包含过当前子集）
TIPS：

1. 通过左移运算符<<来产生二进制数。（不需要声明二进制数）
2. 通过二进制&运算符，来取二进制数的每一位，用于判断子集是否取对应位置的元素。

class Solution {
public:
vector<vector<int>> subsetsWithDup(vector<int>& nums) {
	sort(nums.begin(), nums.end());
	int S = nums.size();
	int N = 1 << S;//二进制数
	vector<vector<int> > res;
	for (int i = 0; i < N; ++i) {
		vector<int> v;
		bool flag = true;
		for (int j = 0; j < S; ++j)
		{
			if (i & (1 << j))//取每一位：不需要声明为二进制，这样就可以！！ 
			{
				if (j != 0 && nums[j] == nums[j - 1] && ((i&(1 << (j - 1))) == 0))//除重
				{
					flag = false;
					break;
				}	
				v.push_back(nums[j]);
			}
		}
		if(flag)
			res.push_back(v);
	}
	return res;
}
};

如果题解对你有启发或者帮助的话，不妨点个赞~
Thanks for your reading!!