原题地址:https://leetcode-cn.com/problems/subsets-ii/submissions/
题目描述:
给定一个可能包含重复元素的整数数组 nums,返回该数组所有可能的子集(幂集)。
说明:解集不能包含重复的子集。
示例:
输入: [1,2,2]
输出:
[
[2],
[1],
[1,2,2],
[2,2],
[1,2],
[]
]
解题方案:
这道题刚开始不知道怎么跟自己过不去非要用动态规划写...规则是先把i-1的子集全部加入,然后再加入i-1的子集中每个集合后面添加一个nums[i],但是动态规划去重太麻烦了,如果有重复的数字,除了把i-1的子集全部加入,如果nums[i]前面有k个数字于它相同,那么还要向i-1子集中含有前k个重复数字的集合中添加num[i]再加入i的子集,自己还是没想出来解法。
这是参考:https://blog.csdn.net/feliciafay/article/details/18977055
用回溯写去重就比较简单了,重复数字中只有第一个要考虑它已经包含了最多数字了,所以后面的都不需要考虑。
代码:
class Solution {
public:
vector<vector<int>> result;
vector<vector<int>> subsetsWithDup(vector<int>& nums) {
sort(nums.begin(), nums.end());
vector<int> temp;
result.push_back(temp);
getAns(0, nums, result, temp);
return result;
}
void getAns(int start, vector<int>& nums, vector<vector<int>> &result, vector<int> temp)
{
if(start == nums.size() - 1)
{
temp.push_back(nums[start]);
result.push_back(temp);
}
else
{
for(int i = start; i < nums.size(); i ++)
{
while(i != 0 && i != start && nums[i] == nums[i - 1])
{
i ++;
}
if(i == nums.size())
break;
temp.push_back(nums[i]);
result.push_back(temp);
getAns(i + 1, nums, result, temp);
temp.pop_back();
}
}
}
};