Python怎么實現二叉樹按層級遍歷

這篇文章主要講解了“Python怎么實現二叉樹按層級遍歷”，文中的講解內容簡單清晰，易于學習與理解，下面請大家跟著小編的思路慢慢深入，一起來研究和學習“Python怎么實現二叉樹按層級遍歷”吧！

Python實現二叉樹按層級遍歷，看起來有點復雜。如下圖，如果一個二叉樹有多層，則從上到下，每層按從左到右遍歷。

每層放在一個隊列，整體返回一個多組隊列的隊列。

看起來復雜，其實實現反而很簡單；不用棧，就用一個節點隊列放當前層級節點，遍歷當前層級節點，讀取值放入一個讀取隊列，讀取當前節點的子節點放入另一個隊列；當當前隊列節點遍歷完成后，使用下一層節點對用，相當于進入下一層，遍歷剛剛生成子節點隊列，如此反復直到節點隊列為空結束。

代碼如下

# Definition for a binary tree node.
# class TreeNode:
#     def __init__(self, x):
#         self.val = x
#         self.left = None
#         self.right = None
class Solution:
    def levelOrder(self, root: TreeNode) -> List[List[int]]:
        traversalList = []
        nodeList = []
        if root != None:
            nodeList.append(root)
            while nodeList != []:
                currentLevelVal = []
                currentNodeList = []
                for node in nodeList:
                    currentLevelVal.append(node.val)
                    if node.left != None:
                        currentNodeList.append(node.left)
                    if node.right != None:
                        currentNodeList.append(node.right)
                traversalList.append(currentLevelVal)
                nodeList = currentNodeList
        return traversalList

感謝各位的閱讀，以上就是“Python怎么實現二叉樹按層級遍歷”的內容了，經過本文的學習后，相信大家對Python怎么實現二叉樹按層級遍歷這一問題有了更深刻的體會，具體使用情況還需要大家實踐驗證。這里是億速云，小編將為大家推送更多相關知識點的文章，歡迎關注！