【leetcode】二叉树的后序遍历

递归

# 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 postorderTraversal(self, root: Optional[TreeNode]) -> List[int]:
        # Solution 1: 递归
        res = []
        def dfs(root:Optional[TreeNode]):
            if root == None:
                return
            dfs(root.left)
            dfs(root.right)
            res.append(root.val)
            
        dfs(root)
        return res   

迭代:和中序遍历、前序遍历略有不同

增加一个辅助变量preNode,记录上一轮访问的节点,来判断当前节点的right是不是上一轮被访问的节点

# 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 postorderTraversal(self, root: Optional[TreeNode]) -> List[int]:
       # Solution 2: 迭代
        res = []
        stack = []
        preNode = None
        while root!=None or stack!=[]:
            while root:
                stack.append(root)
                root = root.left
            root = stack.pop()
            # Note(1)
            if root.right == None or root.right == preNode:
                res.append(root.val)
                preNode = root
                root = None
            else:
            	# Note(2)如果当前节点的右子树还没有被访问过,那么当前节点还要再次入栈
                stack.append(root)
                root = root.right
        return res

猜你喜欢

转载自blog.csdn.net/ningmengzhihe/article/details/127833343