【leetcode】Subsets II (middle) ☆

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],
[]
]

思路：回溯法是肯定的了，我用的深度优先，先把输入的S排序，这样当考虑了前面的数字后，后面的讨论中就不需要再考虑该数字了。需要对每次的子序列判断是否已经重复出现。

代码AC了，但是800ms太慢，贴边过的。

class Solution {
public:
vector<vector<int> > subsetsWithDup(vector<int> &S) {
vector<vector<int>> ans;
vector<int> null;
ans.push_back(null);

if(S.size() == 0)
return ans;

sort(S.begin(), S.end());
DFS(ans, S);
return ans;
}

void DFS(vector<vector<int>> &ans, vector<int> S)
{
if(S.empty())
return;

vector<int> vPreLen = ans.back();
while(!S.empty())
{
vector<int> v = vPreLen;
int cur = S[0];
v.push_back(cur);
S.erase(S.begin());
if(!isalreadyhave(ans, v))
{
ans.push_back(v);
DFS(ans, S);
}
}
return;
}

bool isalreadyhave(vector<vector<int>> ans, vector<int> v)
{
for(int i = 0; i < ans.size(); i++)
{
if(v == ans[i])
return true;
}
return false;
}
};

大神的思路：48ms

在压入答案的时候就保证不会重复，把重复的部分如（5,5,5）看做一个特殊的数，压入时可以压入1个、2个、3个。

把S排序后，按顺序从第一个到最后一个得到该数字加入后，可以得到的新的子序列。每次加入一个新的数字后，新的数字会使得所有之前已经压入的子序列产生新的子序列。

class Solution {
public:
vector<vector<int> > subsetsWithDup(vector<int> &S) {
vector<vector<int> > totalset = {{}};
sort(S.begin(),S.end());
for(int i=0; i<S.size();){
int count = 0; // num of elements are the same
while(count + i<S.size() && S[count+i]==S[i])  count++;
int previousN = totalset.size();
for(int k=0; k<previousN; k++){
vector<int> instance = totalset[k];
for(int j=0; j<count; j++){
instance.push_back(S[i]);
totalset.push_back(instance);
}
}
i += count;
}
return totalset;
}
};