劍指offer-從上往下打印二叉樹

從上往下打印出二叉樹的每一個節點,同層節點從左至右打印。app

# -*- coding:utf-8 -*-
# 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 not root:
            return []
        rootList = []  # 節點列表
        rootList.append(root)
        numList = []  # 節點值列表
        numList.append(root.val)
        for i in rootList:
            if i.left:
                rootList.append(i.left)
                numList.append(i.left.val)
            if i.right:
                rootList.append(i.right)
                numList.append(i.right.val)
        
        return numList

思路是,依次將根節點,左孩子,有孩子放入列表中,而後放入左孩子的左孩子,右孩子;右孩子的左孩子,右孩子.......和人腦讀取的順序是同樣的。spa

相關文章
相關標籤/搜索