解题思路
这道题目给的字符串一定是有效的,所以不用担心不能匹配的情形。题目要求给定的一个有效括号字符串seq,将其分成两个不相交的有效括号字符串 A 和 B,并使这两个字符串的深度最小,也即 max(depth(A), depth(B))的可能取值最小,也就是说要使得两个子序列的括号嵌套深度较大者尽量的小。既然想要两个子序列的嵌套深度中较大者尽量小,那么最好能够让两个子序列的嵌套深度相同。
考虑任意一个原序列中嵌套深度为 d 的合法子序列,我们要想办法把它拆成两半。那么最优的方法肯定是一半嵌套深度为 ⌊d/2⌋ ,一半是 d−⌊d/2⌋ 。这样两个子序列中嵌套深度较大值就是⌈d/2⌉,而其它任何分法都会导致较大值大于它。
要划分出使得最大嵌套深度最小的分组,我们首先得知道如何计算嵌套深度。我们可以通过栈实现括号匹配来计算:首先维护一个栈 s,从左至右遍历括号字符串中的每一个字符:如果当前字符是’(’,就把’(‘压入栈中,此时这个’(‘的嵌套深度为栈的高度;如果当前字符是’)’,此时这个’)‘的嵌套深度为栈的高度,随后再从栈中弹出一个’(’。
下面给出了括号序列 (()(())()) 在每一个字符处的嵌套深度:
括号序列 ( ( ) ( ( ) ) ( ) )
下标编号 0 1 2 3 4 5 6 7 8 9
嵌套深度 1 2 2 2 3 3 2 2 2 1
只要在遍历过程中,我们保证栈内一半的括号属于序列 A,一半的括号属于序列 B,那么就能保证拆分后最大的嵌套深度最小,是当前最大嵌套深度的一半。要实现这样的对半分配,我们只需要把奇数层的 ( 分配给 A,偶数层的 ( 分配给 B 即可。对于上面的例子,我们将嵌套深度为 1 和 3 的所有括号 (()) 分配给 A,嵌套深度为 2 的所有括号 ()()() 分配给 B。
此外,由于在这个问题中,栈中只会存放 (,因此我们不需要维护一个真正的栈,只需要用一个变量模拟记录栈的大小。
具体实现上,用一个和括号字符串等大的数组,它的值为0或者1,表示当前括号被分给第一个字符串还是第二个字符串。我们用一个变量 depth 来表示当前括号的嵌套深度,那么遇到左括号就深度加一,遇到右括号嵌套深度就是当前的 depth,但是遍历完这个右括号之后,深度要减一。最后嵌套深度为偶数的括号位置处标记为 0,奇数的括号位置处标记为 1。
代码
class Solution:
def maxDepthAfterSplit(self, seq: str) -> List[int]:
res = []
depth = 0
for s in seq:
if s=='(':
depth += 1
res.append(depth%2)
elif s==')':
res.append(depth%2)
depth -= 1
return res