剑指offer 从上往下打印二叉树

题目描述:
不分行从上往下打印出二叉树的每个节点,同层节点从左至右打印。例如输入{8,6,10,#,#,2,1},如以下图中的示例二叉树,则依次打印8,6,10,2,1(空节点不打印,跳过),请你将打印的结果存放到一个数组里面,返回。
在这里插入图片描述
示例1

输入:{8,6,10,#,#,2,1}
返回值:[8,6,10,2,1]

思路:
1)采用二叉树的层次遍历法再结合队列的先进先出机制/辅助栈进行解题
对二叉树进行层次遍历,使用栈/队列存储遍历节点,并将该节点存入返回列表,判断该节点是否存在左右子树,若存在左右子树则将该节点的左右子树存入栈/队列,接下来取出栈/队列的第一个元素重复以上动作,直到栈/队列中的节点都存入到返回列表中,则输出返回列表
(2)使用存储节点的链表,控制链表为先进先出,按照题目要求的层序,设置Index存储节点读取的顺序,当链表连接一个新的节点时,将判断该节点是否有左右子节点,并按序添加到链表上
终止条件:由于最后读取的节点没有子节点,所以链表长度将不再增加,终止条件即为循环结束;终止后,遍历节点链表,将val赋值给int数组即可。

Python代码实现:

class Solution:
    def PrintFromTopToBottom(self , root: TreeNode) -> List[int]:
        # write code here
        res=[]
        if not root:
            return []
        # 将根节点入栈
        stack=[root]
        while stack:
            # 获取二叉树每层的元素,搜索其左右子树
            node=stack[0]
            stack=stack[1:]
            # 节点存入列表
            res.append(node.val)
            # 左节点入栈
            if node.left:
                stack.append(node.left)
            # 右节点入栈
            if node.right:
                stack.append(node.right)
        return res
  • 0
    点赞
  • 0
    收藏
    觉得还不错? 一键收藏
  • 0
    评论

“相关推荐”对你有帮助么?

  • 非常没帮助
  • 没帮助
  • 一般
  • 有帮助
  • 非常有帮助
提交
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值