[ LeetCode 102 ] 二叉樹的層序遍歷

天天分享一個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
複製代碼

層序遍歷

層序遍歷比較重要！

第一點，層序自己就是一種 BFS 的思想，做爲最爲基礎的一個題目，校招中不少時候會被問到；

第二點，在解決一些其餘類型的題目的時候，BFS 的思想能夠做爲不少題目的基礎解法，這個在後面的題目中會不少次的用到。

一般解法思路（利用隊列進行輔助）：

1.初始化一個隊列queue，而後將頭結點入隊；

2.隊列 queue 出隊首元素 node，而且打印

3.判斷結點 node 是否有左右孩子。若是有，左右孩子依次入隊，而後執行第 2 點。不然，直接執行第 2 點

實現的代碼

def level_order_traverse(self, head):
    if not head:
        return
    queue = [head]
    while len(queue) > 0:
        tmp = queue.pop(0)
        print(tmp.value, end=" ")
        if tmp.left:
            queue.append(tmp.left)
        if tmp.right:
            queue.append(tmp.right)
複製代碼

而題目中，要求是這樣的形式輸出。即：將每一行做爲一個列表輸出

二叉樹：[3,9,20,null,null,15,7],

    3
   / \
  9  20
    /  \
   15   7
返回其層序遍歷結果：

[
  [3],
  [9,20],
  [15,7]
]
複製代碼

咱麼應該在輸出的部分作一些改變，須要在遍歷到每行的時候，進行單獨保存，以後合併到最終的數組當中。

單獨保存的部分，進行循環遍歷，將下一層的結點更新到另一個數組當中。

看代碼

class Solution(object):
    def levelOrder(self, root):
        res = []
        if not root:
            return res
        queue = [root]
        while queue:
            tmp_queue = []      # 臨時記錄每一層結點
            tmp_res = []        # 臨時記錄每一行的結點值
            for node in queue:
                tmp_res.append(node.val)
                if node.left:
                    tmp_queue.append(node.left)
                if node.right:
                    tmp_queue.append(node.right)
            queue = tmp_queue
            res.append(tmp_res)
        return res
複製代碼

其實就是作了一個很小的操做便可實現

完整代碼

包括測試案例

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

# 樹結點類
class TreeNode(object):
    def __init__(self, val, left=None, right=None):
        self.val = val
        self.left = left
        self.right = right


class Solution(object):
    def levelOrder(self, root):
        res = []
        if not root:
            return res
        queue = [root]
        while queue:
            tmp_queue = []      # 臨時記錄每一層結點
            tmp_res = []        # 臨時記錄每一行的結點值
            for node in queue:
                tmp_res.append(node.val)
                if node.left:
                    tmp_queue.append(node.left)
                if node.right:
                    tmp_queue.append(node.right)
            queue = tmp_queue
            res.append(tmp_res)
        return 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.levelOrder(root))
複製代碼