C++實現(xiàn)LeetCode(90.子集合之二)

更新時間：2021年07月15日 09:35:23 作者：Grandyang

這篇文章主要介紹了C++實現(xiàn)LeetCode(90.子集合之二),本篇文章通過簡要的案例,講解了該項技術的了解與使用,以下就是詳細內容,需要的朋友可以參考下

[LeetCode] 90. Subsets II 子集合之二

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

這道子集合之二是之前那道 Subsets 的延伸，這次輸入數(shù)組允許有重復項，其他條件都不變，只需要在之前那道題解法的基礎上稍加改動便可以做出來，我們先來看非遞歸解法，拿題目中的例子 [1 2 2] 來分析，根據(jù)之前 Subsets 里的分析可知，當處理到第一個2時，此時的子集合為 [], [1], [2], [1, 2]，而這時再處理第二個2時，如果在 [] 和 [1] 后直接加2會產生重復，所以只能在上一個循環(huán)生成的后兩個子集合后面加2，發(fā)現(xiàn)了這一點，題目就可以做了，我們用 last 來記錄上一個處理的數(shù)字，然后判定當前的數(shù)字和上面的是否相同，若不同，則循環(huán)還是從0到當前子集的個數(shù)，若相同，則新子集個數(shù)減去之前循環(huán)時子集的個數(shù)當做起點來循環(huán)，這樣就不會產生重復了，代碼如下：

解法一：

class Solution {
public:
    vector<vector<int>> subsetsWithDup(vector<int> &S) {
        if (S.empty()) return {};
        vector<vector<int>> res(1);
        sort(S.begin(), S.end());
        int size = 1, last = S[0];
        for (int i = 0; i < S.size(); ++i) {
            if (last != S[i]) {
                last = S[i];
                size = res.size();
            }
            int newSize = res.size();
            for (int j = newSize - size; j < newSize; ++j) {
                res.push_back(res[j]);
                res.back().push_back(S[i]);
            }
        }
        return res;
    }
};

整個添加的順序為：

[]
[1]
[2]
[1 2]
[2 2]
[1 2 2]

對于遞歸的解法，根據(jù)之前 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é)點，因為它們是重復的子集，應被拋棄。代碼如下：

解法二：

class Solution {
public:
    vector<vector<int>> subsetsWithDup(vector<int> &S) {
        if (S.empty()) return {};
        vector<vector<int>> res;
        vector<int> out;
        sort(S.begin(), S.end());
        getSubsets(S, 0, out, res);
        return res;
    }
    void getSubsets(vector<int> &S, int pos, vector<int> &out, vector<vector<int>> &res) {
        res.push_back(out);
        for (int i = pos; i < S.size(); ++i) {
            out.push_back(S[i]);
            getSubsets(S, i + 1, out, res);
            out.pop_back();
            while (i + 1 < S.size() && S[i] == S[i + 1]) ++i;
        }
    }
};

整個添加的順序為：

[]
[1]
[1 2]
[1 2 2]
[2]
[2 2]

到此這篇關于C++實現(xiàn)LeetCode(90.子集合之二)的文章就介紹到這了,更多相關C++實現(xiàn)子集合之二內容請搜索腳本之家以前的文章或繼續(xù)瀏覽下面的相關文章希望大家以后多多支持腳本之家！

您可能感興趣的文章: