[LintCode 69] 二叉树的层次遍历 (Python)

2019年4月18日 209次阅读 来源: 愚人国王

题目描述

给出一棵二叉树，返回其节点值的层次遍历（逐层从左往右访问）

样例

给一棵二叉树 {3,9,20,#,#,15,7} ，

  3
 / \ 9  20
  /  \  15   7

返回他的分层遍历结果：

[
  [3],
  [9,20],
  [15,7]
]

代码如下

""" Definition of TreeNode: class TreeNode: def __init__(self, val): self.val = val self.left, self.right = None, None """


class Solution:
    """ @param root: The root of binary tree. @return: Level order in a list of lists of integers """
    def levelOrder(self, root):
        # write your code here
        res = []
        # 如果根节点为空，则返回空列表
        if root is None:
            return res
        # 模拟一个队列储存节点
        q = []
        # 首先将根节点入队
        q.append(root)
        # 列表为空时，循环终止
        while len(q) != 0:
            # 使用列表存储同层节点
            tmp = []
            # 记录同层节点的个数
            length = len(q)
            for i in range(length):
                # 将同层节点依次出队
                r = q.pop(0)
                if r.left is not None:
                    # 非空左孩子入队
                    q.append(r.left)
                if r.right is not None:
                    # 非空右孩子入队
                    q.append(r.right)
                tmp.append(r.val)
            res.append(tmp)
        return res
    原文作者：愚人国王
    原文地址: https://blog.csdn.net/yurenguowang/article/details/76906620
    本文转自网络文章，转载此文章仅为分享知识，如有侵权，请联系博主进行删除。
点赞