牛客网 从上往下打印二叉树

题目:

从上往下打印出二叉树的每个节点,同层节点从左至右打印。

解法:

最直接的想法就树的广度遍历,解题思路参考树的广度优先遍历

# -*- 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 []
        tmp_array=[]
        tmp_array.append(root)
        result_array=[]
        while len(tmp_array)>0:
            cur_node=tmp_array.pop(0)
            result_array.append(cur_node.val)
            if cur_node.left:
                tmp_array.append(cur_node.left)
            if cur_node.right:
                tmp_array.append(cur_node.right)

        return result_array

 

你可能感兴趣的:(数据结构,牛客网)