美文网首页剑指offer-python
面试题22:从上往下打印二叉树

面试题22:从上往下打印二叉树

作者: fighting_css | 来源:发表于2018-07-29 00:05 被阅读0次

【题目描述】
从上往下打印出二叉树的每个节点,同层节点从左至右打印。
【思路】
新加入一个节点时,将该节点的左右子树添加进队列


image.png
image.png

【代码】:

# class TreeNode:
#     def __init__(self, x):
#         self.val = x
#         self.left = None
#         self.right = None
class Solution:
   # 返回从上到下每个节点值列表,例:[1,2,3]
   def PrintFromTopToBottom(self, root):
       # write code here
       if root==None:
           return []
       res = []
       queue = [root]
       #将根加入队列
       while queue:
           #取出队列的头结点
           pNode_list = queue
           queue = []
           for pNode in pNode_list:
               #结果队列加入相应的节点值
               res.append(pNode.val)
               if pNode.left:
                   queue.append(pNode.left)
               if pNode.right:
                   queue.append(pNode.right)
           #将头节点取出后,更新队列
       return res

相关文章

网友评论

    本文标题:面试题22:从上往下打印二叉树

    本文链接:https://www.haomeiwen.com/subject/danamftx.html