-
Notifications
You must be signed in to change notification settings - Fork 1
Expand file tree
/
Copy pathquestion_3.py
More file actions
110 lines (94 loc) · 2.44 KB
/
question_3.py
File metadata and controls
110 lines (94 loc) · 2.44 KB
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
#!/usr/bin/env python
# @Time : 2018/5/12 下午5:52
# @Author : cancan
# @File : question_3.py
# @Function : 对称二叉树
"""
Question:
给定一个二叉树,检查它是否是镜像对称的。
例如,二叉树 [1,2,2,3,4,4,3] 是对称的。
1
/ \
2 2
/ \ / \
3 4 4 3
但是下面这个 [1,2,2,null,3,null,3] 则不是镜像对称的:
1
/ \
2 2
\ \
3 3
Note:
如果你可以运用递归和迭代两种方法解决这个问题,会很加分。
"""
def stringToTreeNode(input):
input = input.strip()
input = input[1:-1]
if not input:
return None
inputValues = [s.strip() for s in input.split(',')]
root = TreeNode(int(inputValues[0]))
nodeQueue = [root]
front = 0
index = 1
while index < len(inputValues):
node = nodeQueue[front]
front = front + 1
item = inputValues[index]
index = index + 1
if item != "null":
leftNumber = int(item)
node.left = TreeNode(leftNumber)
nodeQueue.append(node.left)
if index >= len(inputValues):
break
item = inputValues[index]
index = index + 1
if item != "null":
rightNumber = int(item)
node.right = TreeNode(rightNumber)
nodeQueue.append(node.right)
return root
# Definition for a binary tree node.
class TreeNode(object):
def __init__(self, x):
self.val = x
self.left = None
self.right = None
class Solution(object):
def isSymmetric(self, root):
"""
:type root: TreeNode
:rtype: bool
"""
if not root:
return True
t = []
n = []
def f(node, t, n):
if node.left:
t.append(node.left)
n.append(node.left.val)
else:
n.append(None)
if node.right:
t.append(node.right)
n.append(node.right.val)
else:
n.append(None)
f(root, t, n)
while t:
l = len(n)
for _ in range(l // 2):
if n[_] != n[l - _ - 1]:
return False
p = []
n = []
for i in t:
f(i, p, n)
t = p
return True
if __name__ == "__main__":
a = Solution()
root = stringToTreeNode('[1,2,3,3,null,2,null]')
print(a.isSymmetric(root))