剑指offer-python

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

2018-07-29  本文已影响0人  fighting_css

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


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
上一篇 下一篇

猜你喜欢

热点阅读