[Python]二叉树中序遍历代码以及思路

最新推荐文章于 2023-07-05 11:30:16 发布

清水寺丞

最新推荐文章于 2023-07-05 11:30:16 发布

阅读量1k

点赞数 1

分类专栏：刷题心得文章标签：二叉树中序遍历栈算法数据结构

本文链接：https://blog.csdn.net/m0_56738500/article/details/126735739

版权

刷题心得专栏收录该内容

6 篇文章 0 订阅

订阅专栏

思路：

把中序遍历分为一个有三个阶段的循环，左子节点入栈，没有右子节点出栈，
有右子节点则出栈并且右子节点入栈。

代码：

二叉树节点定义如下

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: Optional[TreeNode]) -> List[int]:
        ans=[]
        if(root==None):
            return []
        
        st=[root]
        while(st):
            root=st[-1]      
            while(root.left!=None):
                st.append(root.left)
                root=root.left
            while(st and st[-1].right==None):
                ans=ans+[st.pop().val]
            if(st):
                right_node=st.pop()
                ans=ans+[right_node.val]
                st.append(right_node.right)