[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 を追加して、前回のラウンドで訪問したノードを記録し、現在のノードの右が前回のラウンドで訪問したノードであるかどうかを判断します

# 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