856. 括号的分数
给定一个平衡括号字符串 S,按下述规则计算该字符串的分数:
https://leetcode-cn.com/problems/score-of-parentheses/
() 得 1 分。
AB 得 A + B 分,其中 A 和 B 是平衡括号字符串。
(A) 得 2 * A 分,其中 A 是平衡括号字符串。
示例 1:
输入: "()"
输出: 1
示例 2:
输入: "(())"
输出: 2
示例 3:
输入: "()()"
输出: 2
示例 4:
输入: "(()(()))"
输出: 6
提示:
S 是平衡括号字符串,且只含有 ( 和 ) 。
2 <= S.length <= 50
class Solution:
def scoreOfParentheses(self, S):
res = 0
depth = 0
for index, i in enumerate(S):
if i == '(':
depth += 1
else:
depth -= 1
if S[index-1] == '(':
res += 2**depth
return res
def test_scoreOfParentheses():
s = Solution()
assert s.scoreOfParentheses('()') == 1
assert s.scoreOfParentheses('(())') == 2
assert s.scoreOfParentheses('()()') == 2
assert s.scoreOfParentheses('(()(()))') == 6