題目描述
給定一組可能包含重複元素的整數數組 nums,返回該數組所有可能的子集(冪集)。
說明: 解集不能包含重複的子集。
示例:
輸入: nums = [1,2,2]
輸出:
[
[2],
[1],
[1,2,2],
[2,2],
[1,2],
[]
]
我的思路
- 是LeetCode-子集的加強版,增加了一些判斷條件,整體的思路沒變。
- 同樣使用回溯法,屬於沒有遞歸基的遞歸,每調用一次遞歸,就輸出當前 temp 中的元素。例如本題,元素彈出的順序爲 1 | 1,2 | 1,2,2 | 1,2 | 2,2 | 2 。遞歸樹如下:
root
/ | \
1 2 2
/ \ |
2 2 2
|
3
- 但我們會發現,1,2 數組有兩個,不符題意。因此每遞歸到一層,應該去掉該層相同的數字,比如例子中該去掉第二層的第二個2。
- 還有一點不同,需要在運算前,將原數組進行排序。
代碼如下:
class Solution {
vector<vector<int>> res;
vector<int> temp;
public:
vector<vector<int>> subsetsWithDup(vector<int>& nums) {
sort(nums.begin(), nums.end());
recur(nums, 0);
return res;
}
void recur(vector<int>& nums, int index){
res.push_back(temp);
for(int i = index; i < nums.size(); i++){
if(i > index && nums[i] == nums[i-1]) continue; //去掉該層相同的數字
temp.push_back(nums[i]);
recur(nums, i + 1);
temp.pop_back();
}
return;
}
};