Leetcode 90. 子集 II dfs

Posted 2021-05-22 itdef

地址 https://leetcode-cn.com/problems/subsets-ii/

给你一个整数数组 nums ，其中可能包含重复元素，请你返回该数组所有可能的子集（幂集）。
解集 不能 包含重复的子集。返回的解集中，子集可以按 任意顺序 排列。

示例 1：
输入：nums = [1,2,2]
输出：[[],[1],[1,2],[1,2,2],[2],[2,2]]

示例 2：
输入：nums = [0]
输出：[[],[0]]

提示：
1 <= nums.length <= 10
-10 <= nums[i] <= 10

解答
对于每个元素 我们有选择或者不选择的方案。
但是需要避免重复的选择 比如对于1 2 2 ，第一个2不选择和第二个2不选择 得到的结果都是 [1,2]
要避免重复的方案就是 一旦不选择2 后面的2 都不必选择。
所以我们还需要将nums进行排序，这样更加方便执行一旦不选择某个数后面相同的数都不选择

class Solution {
public:
	vector<vector<int>> ans;
	void dfs(int idx, vector<int>& v, const vector<int>& nums) {
		if (idx >= nums.size()) {
			ans.push_back(v);
			return;
		}

		v.push_back(nums[idx]);
		dfs(idx + 1, v, nums);
		v.pop_back();

		//当前元素不选择 
		//为避免重复 后面相同元素均不选择
		int val = nums[idx];
		while (idx < nums.size() && nums[idx] == val) { idx++; }
		dfs(idx , v, nums); 
	}

	vector<vector<int>> subsetsWithDup(vector<int>& nums) {
        sort(nums.begin(),nums.end());
		vector<int> v;
		dfs(0, v, nums);
		return ans;
	}
};

我的视频题解空间