-
Notifications
You must be signed in to change notification settings - Fork 1
/
Copy path224.py
46 lines (39 loc) · 1.13 KB
/
224.py
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
class Solution(object):
def calculate(self, s):
"""
:type s: str
:rtype: int
"""
result = 0
sign = 1
number = 0
stack = []
for c in s :
if c.isdigit():
number = number * 10 + (int(c) - int('0'))
elif c == '+' :
result += sign * number
sign = 1
number = 0
elif c == '-':
result += sign * number
sign = -1
number = 0
elif c == '(':
stack.append(result)
stack.append(sign)
result = 0
sign = 1
elif c == ')':
result += sign * number
number = result
sign = stack.pop()
result = stack.pop()
result += sign * number
return result
if __name__ == "__main__":
s = Solution()
print(s.calculate("(1+(4+5+2)-3)+(6+8)"))
print(s.calculate("2 - 1 + 2"))
print(s.calculate("(7)-(0)+(4)"))
#print(s.calculate("(3-(5-(8)-(2+(9-(0-(8-(2))))-(4))-(4)))"))