【算法】【遞歸篇】【樹】第8節:leetcode 102. 二叉樹的層序遍歷(BFS與DFS)

本期任務:介紹leetcode中樹的幾個經典問題的遞歸解法

【算法】【遞歸篇】【樹】第1節:leetcode 100. 相同的樹

【算法】【遞歸篇】【樹】第2節:leetcode 105. 從前序與中序遍歷序列構造二叉樹

【算法】【遞歸篇】【樹】第3節:leetcode 210. 課程表 II

【算法】【遞歸篇】【樹】第4節:leetcode 236. 二叉樹的最近公共祖先

【算法】【遞歸篇】【樹】第5節:leetcode 572. 另一個樹的子樹

【算法】【遞歸篇】【樹】第6節:leetcode 101. 對稱二叉樹

【算法】【遞歸篇】【樹】第7節:leetcode 98. 驗證二叉搜索樹

【算法】【遞歸篇】【樹】第8節:leetcode 102. 二叉樹的層序遍歷


問題來源


102. 二叉樹的層序遍歷
給你一個二叉樹,請你返回其按 層序遍歷 得到的節點值。 (即逐層地,從左到右訪問所有節點)。



示例:
二叉樹:[3,9,20,null,null,15,7],

    3
   / \
  9  20
    /  \
   15   7
返回其層次遍歷結果:

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

大佬解析


代碼

"""
需求:
    給你一個二叉樹,請你返回其按 層序遍歷 得到的節點值。 (即逐層地,從左到右訪問所有節點)。
    
思路1:
    非遞歸法(BFS):使用隊列存儲每一層的節點,按層數依次出隊,若節點的左右孩子存在,則孩子入隊
    
思路2:
    遞歸解法(DFS):先序遍歷,加入表示當前層數的index,聯想回溯思想進行類比
    
"""


# 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):
        """遞歸解法(DFS):先序遍歷,加入表示當前層數的index,聯想回溯思想進行類比"""

        if not root:
            return []

        ans = list()

        def helper(index, node):
            if len(ans) <= index:
                ans.append([])

            # 先序遍歷思路
            ans[index].append(node.val)
            if node.left:
                helper(index + 1, node.left)
            if node.right:
                helper(index + 1, node.right)

        helper(0, root)
        return ans

    def levelOrder1(self, root: TreeNode):
        # 非遞歸法(BFS):使用隊列存儲每一層的節點,按層數依次出隊,若節點的左右孩子存在,則孩子入隊
        if not root:
            return []

        ans = list()
        queue = [root]
        while queue:
            # 獲取當前隊列的長度,這個長度相當於 當前這一層的節點個數
            size = len(queue)
            res = list()
            for i in range(size):
                # 將隊列中的元素都拿出來(也就是獲取這一層的節點),放到臨時list中
                # 如果節點的左/右子樹不爲空,也放入隊列中
                node = queue.pop(0)
                res.append(node.val)
                if node.left:
                    queue.append(node.left)
                if node.right:
                    queue.append(node.right)
            ans.append(res)  # 將臨時list加入最終返回結果中
        return ans
發表評論
所有評論
還沒有人評論,想成為第一個評論的人麼? 請在上方評論欄輸入並且點擊發布.
相關文章