LeetCode: Subsets II [091]

【题目】

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，返回所有可能的子集合

1. 要求子集中的元素必须非递减排列

2. 返回的结果中不能出现重复的集合

【思路】

与Subsets相同，先对集合排序，保证得到的子集非递减，然后递归的求出长度分别为0,1,2,3,4..n的所有组合。多了一步就是注意组合的去重。

【代码】

class Solution {
public:

void getCombinations(vector<vector<int> >&result, vector<int>&s, vector<int>combination, int index2add, int kth, int k){
//将s中的第index2add位上的数作为组合中的第kth个数
combination.push_back(s[index2add]);
if(kth==k){
result.push_back(combination);
return;
}

for(int i=index2add+1; i+(k-kth-1)<s.size(); i++){
if(i!=index2add+1 && s[i]==s[i-1])continue; //去重
getCombinations(result, s, combination, i, kth+1, k);
}
}

vector<vector<int> > getSubsetK(vector<int>&s, int k){
// 求长度为k的子集
vector<vector<int> >result;
vector<int> combination;
for(int i=0; i+(k-1)<s.size(); i++){
if(i!=0 && s[i]==s[i-1])continue;   //排重
getCombinations(result, s, combination, i, 1, k);
}
return result;
}

vector<vector<int> > subsetsWithDup(vector<int> &S) {
vector<vector<int> >result;
vector<int> emptySet;   //一定有个空集
result.push_back(emptySet);

int size=S.size();
if(size==0)return result;

sort(S.begin(), S.end());

for(int k=1; k<=size; k++){
//分别求出长度为1,2,3,...size的子集
vector<vector<int> >subsets=getSubsetK(S, k);
result.insert(result.end(), subsets.begin(), subsets.end());
}
return result;
}
};