【Leetcode笔记】二叉树的统一迭代法

最新推荐文章于 2024-07-25 19:34:38 发布

zyzcuczyu

最新推荐文章于 2024-07-25 19:34:38 发布

阅读量29

点赞数

文章标签： leetcode 笔记算法

本文链接：https://blog.csdn.net/zyzcuczyu/article/details/131076110

版权

Leetcode原题链接：
二叉树的前序遍历
 二叉树的中序遍历
 二叉树的后序遍历

一、思路

递归遍历代码非常统一，只需要改变顺序即可，而迭代遍历中序的代码修改较多，现提出较为统一的迭代法
做标记：不论是否处理过，均进行入栈，但处理过的元素入栈后，再入栈一个null值作为标记，当遍历到null值时，继续pop出其前方处理过的元素，加入结果列表

文字描述有点难理解，可以做一个1~7的二叉树手动试试

二、代码

# Definition for a binary tree node.
# class TreeNode:
#     def __init__(self, val=0, left=None, right=None):
#         self.val = val
#         self.left = left
#         self.right = right
class Solution:
    def preorderTraversal(self, root: TreeNode) -> List[int]:
        result = []
        st= []
        if root:
            st.append(root)
        while st:
            node = st.pop()
            if node != None:
                if node.right: 				#右
                    st.append(node.right)
                if node.left: 				#左
                    st.append(node.left)		
                st.append(node) 			#中
                st.append(None)
            else:
                node = st.pop()
                result.append(node.val)
        return result