LeetCode刷题日记2022-3-10/589. N 叉树的前序遍历-递归&栈方法

题目描述

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

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

示例 1:
在这里插入图片描述

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

示例 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]
输出:[1,2,3,6,7,11,14,4,8,12,5,9,13,10]

提示:

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

题解思路

方法一-递归

我们采用递归的方法对N叉树进行遍历然后将节点值存储到数组中

方法二-栈方法

我们遍历根节点 并且将根节点的 子树以倒序的形式存储进栈中

然后依次弹出 直到栈空

题解代码

递归

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

栈方法

class Solution:
    def preorder(self, root: 'Node') -> List[int]:
        if not root:
            return []
        ans=[]
        q=[root]

        while q:
            
            node=q.pop()
            ans.append(node.val)
            q.extend(node.children[::-1])
        
        return ans
  • append添加元素是将节点的孩子节点作为一个整体添加到栈里
  • extend则是将子节点依次迭代添加进入栈
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值