LeetCode(90):子集 II
Medium!
題目描述:
給定一個可能包含重復元素的整數數組?nums,返回該數組所有可能的子集(冪集)。
說明:解集不能包含重復的子集。
示例:
輸入: [1,2,2] 輸出: [[2],[1],[1,2,2],[2,2],[1,2],[] ]解題思路:
這道子集合之二是之前那道?Subsets 子集合?的延伸,這次輸入數組允許有重復項,其他條件都不變,只需要在之前那道題解法的基礎上稍加改動便可以做出來,我們先來看非遞歸解法,拿題目中的例子[1 2 2]來分析,根據之前?Subsets 子集合?里的分析可知,當處理到第一個2時,此時的子集合為[], [1], [2], [1, 2],而這時再處理第二個2時,如果在[]和[1]后直接加2會產生重復,所以只能在上一個循環(huán)生成的后兩個子集合后面加2,發(fā)現了這一點,題目就可以做了,我們用last來記錄上一個處理的數字,然后判定當前的數字和上面的是否相同,若不同,則循環(huán)還是從0到當前子集的個數,若相同,則新子集個數減去之前循環(huán)時子集的個數當做起點來循環(huán),這樣就不會產生重復了。
C++解法一:
1 class Solution { 2 public: 3 vector<vector<int>> subsetsWithDup(vector<int> &S) { 4 if (S.empty()) return {}; 5 vector<vector<int>> res(1); 6 sort(S.begin(), S.end()); 7 int size = 1, last = S[0]; 8 for (int i = 0; i < S.size(); ++i) { 9 if (last != S[i]) { 10 last = S[i]; 11 size = res.size(); 12 } 13 int newSize = res.size(); 14 for (int j = newSize - size; j < newSize; ++j) { 15 res.push_back(res[j]); 16 res.back().push_back(S[i]); 17 } 18 } 19 return res; 20 } 21 };整個添加的順序為:
[]
[1]
[2]
[1 2]
[2 2]
[1 2 2]
?
對于遞歸的解法,根據之前?Subsets 子集合?里的構建樹的方法,在處理到第二個2時,由于前面已經處理了一次2,這次我們只在添加過2的[2] 和 [1 2]后面添加2,其他的都不添加,那么這樣構成的二叉樹如下圖所示:
?
[] / \ / \ / \[1] []/ \ / \/ \ / \ [1 2] [1] [2] []/ \ / \ / \ / \[1 2 2] [1 2] X [1] [2 2] [2] X []?
代碼只需在原有的基礎上增加一句話,while (S[i] == S[i + 1]) ++i; 這句話的作用是跳過樹中為X的葉節(jié)點,因為它們是重復的子集,應被拋棄。
C++解法二:
1 class Solution { 2 public: 3 vector<vector<int>> subsetsWithDup(vector<int> &S) { 4 if (S.empty()) return {}; 5 vector<vector<int>> res; 6 vector<int> out; 7 sort(S.begin(), S.end()); 8 getSubsets(S, 0, out, res); 9 return res; 10 } 11 void getSubsets(vector<int> &S, int pos, vector<int> &out, vector<vector<int>> &res) { 12 res.push_back(out); 13 for (int i = pos; i < S.size(); ++i) { 14 out.push_back(S[i]); 15 getSubsets(S, i + 1, out, res); 16 out.pop_back(); 17 while (i + 1 < S.size() && S[i] == S[i + 1]) ++i; 18 } 19 } 20 };整個添加的順序為:
[]
[1]
[1 2]
[1 2 2]
[2]
[2 2]
轉載于:https://www.cnblogs.com/ariel-dreamland/p/9159492.html
總結
以上是生活随笔為你收集整理的LeetCode(90):子集 II的全部內容,希望文章能夠幫你解決所遇到的問題。
- 上一篇: 奥运冠军邓琳琳个人简历(运动员邓琳琳的励
- 下一篇: 在idea下开发出现404错误