LeetCode刷题日记2022-3-12/590. N 叉树的后序遍历-递归&迭代

题目描述

给定一个 n 叉树的根节点 root ，返回 其节点值的 后序遍历 。

n 叉树 在输入中按层序遍历进行序列化表示，每组子节点由空值 null 分隔（请参见示例）。

示例 1：

输入：root = [1,null,3,2,4,null,5,6]
输出：[5,6,3,2,4,1]

示例 2：

输入：root = [1,null,2,3,4,5,null,null,6,7,null,8,null,9,10,null,null,11,null,12,null,13,null,null,14]
输出：[2,6,14,11,7,3,12,8,4,13,9,10,5,1]

提示：

• 节点总数在范围 [0, 104] 内
• 0 <= Node.val <= 104
• n 叉树的高度小于或等于 1000

题解代码

递归

class Solution:
    def postorder(self, root: 'Node') -> List[int]:
        ans=[]


        def dfs(node):
            if not node:
                return None
            
            for ch in node.children:
                dfs(ch)
            
            ans.append(node.val)
        dfs(root)
        return ans

迭代

class Solution:
    def postorder(self, root: 'Node') -> List[int]:
        stack,ans=[root],[]

        while stack:
            node=stack.pop()
            if node is not None:
                if type(node)==Node:
                    stack.append(node.val)
                    stack.extend(node.children[::-1])
                else:
                    ans.append(node)
        return ans