二叉树的前序遍历、中序遍历、后序遍历（递归、迭代实现方式）

最新推荐文章于 2024-10-12 20:00:42 发布

小轩精益求精

最新推荐文章于 2024-10-12 20:00:42 发布

阅读量143

点赞数

文章标签：二叉树栈数据结构

本文链接：https://blog.csdn.net/qq_39098713/article/details/114788795

版权

1、递归（分治法）

class Solution:
    def Traversal(self, root: TreeNode) -> List[int]:
    	if not root:
    		return []
    	# 前序遍历
    	return [root.val] + self.Traversal(root.left) + self.Traversal(root.right)
    	# 中序遍历
    	return self.Traversal(root.left) + [root.val] + self.Traversal(root.right)
    	# 后序遍历
    	return self.Traversal(root.left) + self.Traversal(root.right) + [root.val]

2、非递归（迭代法）

class Solution:
    def Traversal(self, root: TreeNode) -> List[int]:
        res = []
        stack = []
        cur = root
		# 前序遍历
		while stack or cur:
            while cur:
                res.append(cur.val)
                stack.append(cur)
                cur = cur.left
            cur = stack.pop()
            cur = cur.right
        return res
		# 中序遍历
		while stack or cur:
            while cur:
                stack.append(cur)
                cur = cur.left
            cur = stack.pop()
            res.append(cur.val)
            cur = cur.right
        return res
       # 后序遍历
        while stack or cur:
            while cur:
                res.append(cur.val)
                stack.append(cur)
                cur = cur.right
            cur = stack.pop()
            cur = cur.left
        return res[::-1]