leetcode1111

 1 class Solution:
 2     def maxDepthAfterSplit(self, seq: str) -> 'List[int]':
 3         n = len(seq)
 4         stack = []
 5         res = [0] * n
 6         odd = True
 7         for i in range(n):
 8             cur = seq[i]
 9             if len(stack) == 0 or cur == '(':
10                 stack.append((i,cur))
11             else:
12                 top = stack.pop(-1)
13                 if odd:
14                     res[top[0]] = 1
15                     res[i] = 1            
16                 odd = not odd
17         return res

思路:本题使用贪心思想,使用一个stack存储括号。

遍历到")"括号时,则进行分配,要保持A与B的平衡,只需要轮流分配给A和B即可。

举例来说:"( ( ( ) ( ) ) )",遍历此字符串,

i = 0, cur = '(',入栈

i = 1, cur = '(',入栈

i = 2, cur = '(',入栈

i = 3, cur = ')',栈顶必然是'(',则将栈顶和当前两个位置分配给串B(标记1),栈顶出栈。

i = 4, cur = '(',入栈

i = 5,cur = ')',栈顶必然是'(',因为上一次分配给了串B,因此本次将栈顶和当前两个位置分配给串A(标记0),站定出栈。

i = 6,cur = ')',分配给A

i = 7,cur = ')',分配给B

猜你喜欢

转载自www.cnblogs.com/asenyang/p/11146265.html