[ LeetCode 107 ] 二叉树的层序遍历 II

天天分享一个LeetCode题目

天天 5 分钟，一块儿进步！

LeetCode 层序遍历，地址：leetcode-cn.com/problems/bi…

树结点类

class TreeNode(object):
    def __init__(self, val, left=None, right=None):
        self.val = val
        self.left = left
        self.right = right
复制代码

层次遍历

仍是和以前层次遍历一样的思路，须要将每一层的遍历结点做为单独的列表放置到最后的结果列表中去

以前普通的层次遍历思路能够看这里：mp.weixin.qq.com/s/4skiCYIhc…

解法思路（利用队列）：

1.初始化一个队列queue，而后将头结点入队，初始化一个结果集 res；

2.再初始化一个临时队列 level_queue 和 一个列表 level_res，循环访问队列 queue 中元素，并判断其左右孩子，若是有孩子结点依次将孩子结点入队level_queue，而且将当前结点入列表 level_res

3.将 level_queue 赋值给 queue，将 level_res 插入到结果集 res。完成一层的计算。随后再执行第 2 点

4.打印结果集 res

实现的代码

def levelOrderBottom(self, root):
    res = []
    queue = collections.deque()
    queue.appendleft(root)
    while(queue):
        level_queue = []   # 记录下一层的结点值
        level_res = []     # 记录当前层的结点值
        for node in queue:
            if node.left:
                level_queue.append(node.left)
            if node.right:
                level_queue.append(node.right)
            level_res.append(node.val)
        # 将下一层的结点都入到queue中，进行下一层的遍历
        queue = level_queue
        res.insert(0, level_res)
    return list(res)
复制代码

我的感受很巧妙，可是四个初始化定义须要好好熟悉一下

完整代码

# -*- coding:utf-8 -*-
# !/usr/bin/env python

import collections

# 树结点类
class TreeNode(object):
    def __init__(self, val, left=None, right=None):
        self.val = val
        self.left = left
        self.right = right

class Solution(object):
    def levelOrderBottom(self, root):
        res = []
        queue = collections.deque()
        queue.appendleft(root)
        while(queue):
            level_queue = []   # 记录下一层的结点值
            level_res = []     # 记录当前层的结点值
            for node in queue:
                if node.left:
                    level_queue.append(node.left)
                if node.right:
                    level_queue.append(node.right)
                level_res.append(node.val)
            # 将下一层的结点都入到queue中，进行下一层的遍历
            queue = level_queue
            res.insert(0, level_res)
        return list(res)


if __name__ == "__main__":
    # 新建节点
    root = TreeNode('A')
    node_B = TreeNode('B')
    node_C = TreeNode('C')
    node_D = TreeNode('D')
    node_E = TreeNode('E')
    node_F = TreeNode('F')
    node_G = TreeNode('G')
    node_H = TreeNode('H')
    node_I = TreeNode('I')
    # 构建二叉树
    # A
    # / \
    # B C
    # / \ / \
    # D E F G
    # / \
    # H I
    root.left, root.right = node_B, node_C
    node_B.left, node_B.right = node_D, node_E
    node_C.left, node_C.right = node_F, node_G
    node_D.left, node_D.right = node_H, node_I

    s = Solution()
    print(s.levelOrderBottom(root))
复制代码