-
Notifications
You must be signed in to change notification settings - Fork 0
/
Copy pathtree.py
86 lines (75 loc) · 2.32 KB
/
tree.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
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
import sys
import threading
sys.setrecursionlimit(10**6)
threading.stack_size(2**27)
class TreeOrders:
def read(self):
path=r"C:\Users\ASUS\Downloads\21"
file=open(path)
c=0
self.key,self.left,self.right=[],[],[]
for i in file:
if c==0: self.n=int(i.split("\n")[0])
else:
self.key.append(int(i.split(" ")[0].split("\n")[0]))
self.left.append(int(i.split(" ")[1].split("\n")[0]))
self.right.append(int(i.split(" ")[2].split("\n")[0]))
c+=1
def _inOrder_(self, node):
if node == -1:
return
self._inOrder_(self.left[node])
self.result.append(self.key[node])
self._inOrder_(self.right[node])
def inOrder(self):
self.result = []
self._inOrder_(0)
return self.result
def _preOrder_(self, node):
if node == -1:
return
self.result.append(self.key[node])
self._preOrder_(self.left[node])
self._preOrder_(self.right[node])
def preOrder(self):
self.result = []
self._preOrder_(0)
return self.result
def _postOrder_(self, node):
if node == -1:
return
self._postOrder_(self.left[node])
self._postOrder_(self.right[node])
self.result.append(self.key[node])
def postOrder(self):
self.result = []
self._postOrder_(0)
return self.result
def main():
tree = TreeOrders()
tree.read()
path=r"C:\Users\ASUS\Downloads\21.a"
file=open(path)
c=0
for i in file:
str=i.split("\n")[0].split(" ")
if c==0:
ans=tree.inOrder()
f=0
for i in range(len(ans)):
if ans[i]!=int(str[i]): f=1
if f==0: print("Success")
elif c==1:
ans=tree.preOrder()
f=0
for i in range(len(ans)):
if ans[i]!=int(str[i]): f=1
if f==0: print("Success")
elif c==2:
ans=tree.postOrder()
f=0
for i in range(len(ans)):
if ans[i]!=int(str[i]): f=1
if f==0: print("Success")
c+=1
threading.Thread(target=main).start()