1. 程式人生 > 實用技巧 >1111. 有效括號的巢狀深度

1111. 有效括號的巢狀深度

1111. 有效括號的巢狀深度

有效括號字串 定義:對於每個左括號,都能找到與之對應的右括號,反之亦然。詳情參見題末「有效括號字串」部分。

巢狀深度 depth 定義:即有效括號字串巢狀的層數,depth(A) 表示有效括號字串 A 的巢狀深度。詳情參見題末「巢狀深度」部分。

有效括號字串型別與對應的巢狀深度計算方法如下圖所示:

給你一個「有效括號字串」 seq,請你將其分成兩個不相交的有效括號字串,A 和B,並使這兩個字串的深度最小。

不相交:每個 seq[i] 只能分給 A 和 B 二者中的一個,不能既屬於 A 也屬於 B 。
A 或 B 中的元素在原字串中可以不連續。
A.length + B.length = seq.length

深度最小:max(depth(A), depth(B))的可能取值最小。
劃分方案用一個長度為 seq.length 的答案陣列 answer 表示,編碼規則如下:

answer[i] = 0,seq[i] 分給 A 。
answer[i] = 1,seq[i] 分給 B 。
如果存在多個滿足要求的答案,只需返回其中任意 一個 即可。

示例 1:

  輸入:seq = "(()())"
  輸出:[0,1,1,1,1,0]
示例 2:

  輸入:seq = "()(())()"
  輸出:[0,0,0,1,1,0,1,1]
解釋:本示例答案不唯一。
按此輸出 A = "()()", B = "()()", max(depth(A), depth(B)) = 1,它們的深度最小。

像 [1,1,1,0,0,1,1,1],也是正確結果,其中 A = "()()()", B = "()", max(depth(A), depth(B)) = 1 。

方法一:用棧進行括號匹配思路及演算法

要求劃分出使得最大巢狀深度最小的分組,我們首先得知道如何計算巢狀深度。我們可以通過棧實現括號匹配來計算:

維護一個棧 s,從左至右遍歷括號字串中的每一個字元:

如果當前字元是 (,就把 ( 壓入棧中,此時這個 ( 的巢狀深度為棧的高度;

如果當前字元是 ),此時這個 ) 的巢狀深度為棧的高度,隨後再從棧中彈出一個 (。

下面給出了括號序列 (()(())()) 在每一個字元處的巢狀深度:

知道如何計算巢狀深度,問題就很簡單了:只要在遍歷過程中,我們保證棧內一半的括號屬於序列 A,一半的括號屬於序列 B,那麼就能保證拆分後最大的巢狀深度最小,是當前最大巢狀深度的一半。要實現這樣的對半分配,我們只需要把奇數層的 ( 分配給 A,偶數層的 ( 分配給 B 即可。對於上面的例子,我們將巢狀深度為 1 和 3 的所有括號 (()) 分配給 A,巢狀深度為 2 的所有括號 ()()() 分配給 B。

此外,由於在這個問題中,棧中只會存放 (,因此我們不需要維護一個真正的棧,只需要用一個變數模擬記錄棧的大小。

程式碼:

class Solution{
public:
    vector<int >maxDepthAfterSplit(string seq){
        int depth=0,len=seq.length();
        vector<int >ans;
        for (int i = 0; i < len; i++)
        {
             if (seq[i] == '(') {
                ans.push_back((++depth)&1);
            } else {
                ans.push_back((depth--)&1);
            }
        }
        return ans;
    }
};

複雜度分析

時間複雜度:O(n),其中 n 為字串的長度。我們只需要遍歷括號字串一次。

空間複雜度:O(1)。除答案陣列外,我們只需要常數個變數。

方法二:找規律

思路及演算法

我們還是使用上面的例子(()(())()),但這裡我們把()的巢狀深度分成兩行:

有沒有發現什麼規律?

左括號 ( 的下標編號與巢狀深度的奇偶性相反,也就是說:

下標編號為奇數的 (,其巢狀深度為偶數,分配給 B;

下標編號為偶數的 (,其巢狀深度為奇數,分配給 A。

右括號 ) 的下標編號與巢狀深度的奇偶性相同,也就是說:

下標編號為奇數的 ),其巢狀深度為奇數,分配給 A;

下標編號為偶數的 ),其巢狀深度為偶數,分配給 B。

這樣以來,我們只需要根據每個位置是哪一種括號以及該位置的下標編號,就能確定將對應的對應的括號分到哪個組了。

對此規律感興趣的同學的同學可以閱讀下面的證明部分,若不感興趣,可以直接跳到程式碼部分。

證明

對於字串中的任意一個左括號 (,它的下標編號為 x,巢狀深度為 y。如果它之有 l 個左括號和 r 個右括號,那麼根據巢狀深度的定義,有:

y = l - r + 1
y=l−r+1

下標編號與 l 和 r 的關係也可以直接得到,注意下標編號從 0 開始:

x = l + r
x=l+r

由於 l - r和 l + r 同奇偶,因此 l - r + 1(即 y)和 l + r(即 x)的奇偶性相反。

對於字串中的任意一個右括號 ),它的下標編號為 x,巢狀深度為 y。如果它之有 l 個左括號和 r 個右括號,那麼根據巢狀深度的定義,有:

y = l - r
y=l−r

下標編號與 l 和 r 的關係也可以直接得到,注意下標編號從 0 開始:

x = l + r
x=l+r

因此 y 和 x 的奇偶性相同。

class Solution {
public:
    vector<int> maxDepthAfterSplit(string seq) {
        vector<int> ans;
        for (int i = 0; i < (int)seq.size(); ++i) {
            ans.push_back(i & 1 ^ (seq[i] == '('));
        }
        return ans;
    }
};

複雜度分析

時間複雜度:O(n),其中 n 為字串的長度。我們只需要遍歷括號字串一次。

空間複雜度:O(1)。除答案陣列外,我們只需要常數個變數。