blob: b5f58c42d112d0a159532a5e593918c4dfbd096f (
plain)
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
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
|
import sys
import globals
def toString (node):
if node == None:
return ''
chars = '('
if type(node.left) == type(0):
chars += "%d"%node.left
else:
chars += toString(node.left)
chars += node.op
if type(node.right) == type(0):
chars += "%d"%node.right
else:
chars += toString(node.right)
chars += ")"
return chars
class Node(object):
def __init__ (self):
self.left = None
self.right = None
self.parent = None
self.op = None
class ExpParser(object):
def __init__ (self, tokens):
self.tokens = tokens
def jumpToRoot (self):
while self.ptr.parent != None:
self.ptr = self.ptr.parent
def build (self):
self.ptr = Node()
for token in self.tokens:
if token in '+-':
if self.ptr.left == None:
raise globals.ParseError ('')
if self.ptr.right == None:
self.ptr.op = token
else:
self.jumpToRoot()
self.ptr.parent = Node()
self.ptr.parent.left = self.ptr
self.ptr = self.ptr.parent
self.ptr.op = token
elif token in '*/':
if self.ptr.left == None:
raise globals.ParseError ('')
elif self.ptr.right == None:
self.ptr.op = token
else:
num = self.ptr.right
self.ptr.right = Node()
self.ptr.right.parent = self.ptr
self.ptr.right.left = num
self.ptr.right.op = token
self.ptr = self.ptr.right
elif token == '(':
if self.ptr.left == None:
self.ptr.left = Node()
self.ptr.left.parent = self.ptr
self.ptr = self.ptr.left
elif self.ptr.right == None:
self.ptr.right = Node()
self.ptr.right.parent = self.ptr
self.ptr = self.ptr.right
else:
raise globals.ParseError ('')
elif token == ')':
if self.ptr.left == None:
raise globals.ParseError ('')
elif self.ptr.right == None:
raise globals.ParseError ('')
elif self.ptr.parent == None:
pass
else:
self.ptr = self.ptr.parent
else:
num = int(token)
if self.ptr.left == None:
self.ptr.left = num
elif self.ptr.right == None:
self.ptr.right = num
else:
raise globals.ParseError ('')
def dumpTree (self):
self.jumpToRoot()
print toString(self.ptr)
|