Leetcode-94 二叉树的中序遍历的三种方式-迭代、递归、莫里斯，Python实现

相关内容

• Leercode94 网址
• Leetcode-94 二叉树的中序遍历
• Leetcode-144 二叉树的前序遍历
• Leercode-145 二叉树的后序遍历
• 莫里斯算法python实现

给定一个二叉树，返回它的中序遍历。
示例:

具体的思路和解题方法见代码注释

# Definition for a binary tree node.
# class TreeNode:
#     def __init__(self, x):
#         self.val = x
#         self.left = None
#         self.right = None
class Solution:
	'''
	递归法
	递归方法是重复的使用一个函数，对于一个非空的节点
	按照左子树、当前值、右子树的顺序的依次存入结果
	'''
	def inorderTraversal(self, root: TreeNode) -> List[int]:
		result = []
	    def _inorderTraversal(root):
            if root:
                _inorderTraversal(root.left)
                result.append(root.val)
                _inorderTraversal(root.right)
	    _inorderTraversal(root)
	    return result


	'''
	迭代法
	迭代法中需要维护一个堆栈，每次当前值不为空的时候入栈
	当得到树的最左边的节点时（叶子节点的空左子树），
	按照堆栈退栈回到对应的上一层节点(最左边的左子树)，
	并打印当前节点值，并将当前值转到右子树上。
	直到当前值为空（树的右子树的最右边的一颗树的空左叶子节点）
	和当前堆栈为空（树的右子树的最右边的一颗树的叶子）时，得到最右侧的一个叶子
	'''
    def inorderTraversal(self, root: TreeNode) -> List[int]:
        result = []
        stack = []
        current = root
        while stack or current:
            while current:
                if current:
                    stack.append(current)
                    current = current.left
            current = stack.pop()
            result.append(current.val)
            current = current.right
        return result

	'''
	莫里斯算法
	莫里斯算法是将每个节点的中序的前驱节点的空右子树的空内存
	(左子树的最右叶子节点的空右子树的空右子树的空内存）指向该节点，
	作为从中序前驱节点返回上一层的途径。
	和之前的迭代法相比较，迭代法需要一个堆栈依次记录从上层到下层的记录，
	当到达最低层的节点时通过出栈的方式获得上层节点。
	而莫里斯算法，利用之前设置的空内存从底层节点返回上层节点。
	'''
    def inorderTraversal(self, root: TreeNode) -> List[int]:
        result = []
        current = root  # 当前的节点，从根节点开始
        while current:
        	# 如果当前节点没有左子树，那么就输出该节点，并且转到右子树上
        	# 因为，中序遍历，左中右，没有了左子树之后，就输出中，然后转到右子树
            if not current.left:  
                result.append(current.val)
                current = current.right
            else:
            	# 当有左子树时，对于当前的节点，找到它的中序前驱节点
            	# 中序前驱节点是当前节点的左子树的最右叶子节点
                pre = current.left 
                # 找到当前节点的中序前驱，并且判断它的右子树是不是空
                while pre.right and pre.right != current:
                    pre = pre.right
                # 根节点的中序前驱节点是个叶子节点，叶子节点的右子树应该是None
                # 利用这个None的空内存指向该根节点
                # 通过该前驱节点的右子树是不是空 判断 是不是第一次访问该根节点
                # 如果该前驱节点的右子树是空，那么是第一次访问根节点
                # 如果该前驱节点的右子树已经指向了根节点，那么该根节点已经是第二次访问
                if not pre.right:
                	# 如果是第一次访问根节点，第一次找到中序前序
                	# 那么就将中序前驱(叶子)右子树的空内存指向该根节点
					# 并且当前节点指向左子树
					# 指向左子树的原因是已经记录了从底层节点返回上层的方式
                    pre.right = current
                    current = current.left
                else:
                	# 如果现在叶子节点的右子树的内存已经指向根节点了
                	# 那么就将该内存设置为空，回到树之前的样子
                	# 并且打印根节点的值
                	# 因为此时根节点已经是第二次访问了，该根节点下面的节点已经都被打印了
                	# 按照左中右的顺序，根节点的左子树下面的节点都被打印了，那么就到了对应的根节点【中】上，打印【中】，
                	# 然后转到根节点的右子树上
                    pre.right = None
                    result.append(current.val)
                    current =current.right
        return result