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

每天分享一个LeetCode题目

每天 5 分钟，一起进步！

LeetCode 层序遍历，地址：https://leetcode-cn.com/problems/binary-tree-level-order-traversal-ii/

树结点类

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

层次遍历

还是和之前层次遍历同样的思路，需要将每一层的遍历结点作为单独的列表放置到最后的结果列表中去

之前普通的层次遍历思路可以看这里：https://mp.weixin.qq.com/s/4skiCYIhc9mKEBmNkot8wA

解法思路（利用队列）：

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))