中序遍历非递归的要点:
由于要遵循中序遍历的左节点、根节点、右节点的输出规律,因此需要:
(1)、在结点不为空或栈不为空的情况下,从最顶部结点到最左结点依次开始入栈直至结点为空。
(2)、当(root = root.left)指向结点为空时,栈最上面的结点出栈并加入空列表,root指向出栈元素的右节点。若出栈元素的右节点不为空则执行(1),若为空则重复执行(2)。最后返回列表
代码如下:
class TreeNode:
def __init__(self, val = 0, left = None, right = None):
self.val = val
self.left = left
self.right = right
class Solution:
def inorderTraversal(self, root:TreeNode):
if not root:
return []
stack = []
res = []
while stack or root:
if root:
stack.append(root)
root = root.left
else:
cur = stack.pop()
res.append(cur.val)
root = cur.right
return res
node = TreeNode(1)
node.left = TreeNode(2)
node.right = TreeNode(3)
a = Solution()
b = a.inorderTraversal(node)
print(b)