【树-简单】590. N 叉树的后序遍历

最新推荐文章于 2024-06-07 14:33:32 发布

菜菜2022

最新推荐文章于 2024-06-07 14:33:32 发布

阅读量153

点赞数

分类专栏：刷题 # leetcode 文章标签： leetcode

本文链接：https://blog.csdn.net/kz_java/article/details/119888026

版权

leetcode 同时被 2 个专栏收录

714 篇文章 5 订阅

订阅专栏

刷题

643 篇文章 5 订阅

订阅专栏

【题目】
给定一个 N 叉树，返回其节点值的后序遍历。

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]

提示：

N 叉树的高度小于或等于 1000
节点总数在范围 [0, 10^4] 内

【代码】
在这里插入图片描述

class Solution:
    def dfs(self,root):
        if not root:
            return
        for x in root.children:
            self.dfs(x)
        self.ans.append(root.val)
    def postorder(self, root: 'Node') -> List[int]:
        self.ans=[]
        self.dfs(root)
        return self.ans

【方法2】
在这里插入图片描述

class Solution(object):
    def postorder(self, root):
        if root is None:
            return []
        stack, output = [root, ], []
        while stack:
            root = stack.pop()
            if root:
                output.append(root.val)
            stack.extend(root.children)
        return output[::-1]

【方法3】真正的n叉树的后序遍历
在这里插入图片描述

class Solution:
    def postorder(self, root: 'Node') -> List[int]:
        value, s = [], []
        cur = root
        while s or cur:
            while cur:
                s.append((cur, 0))
                if cur.children:
                    cur = cur.children[0]
                else:
                    cur = None
            cur, c = s.pop()
            if not cur.children or c >= len(cur.children):
                value.append(cur.val)
                cur = None
            else:
                c += 1
                s.append((cur, c))
                if c < len(cur.children):
                    cur = cur.children[c]
                else:
                    cur = None
        return value