leetcode学习笔记——二叉树的层序遍历(102,199,637)

2023好运来

已于 2022-03-14 16:30:53 修改

阅读量376

点赞数

分类专栏：学习记录文章标签： leetcode

于 2022-03-14 15:24:29 首次发布

本文链接：https://blog.csdn.net/qq_43656500/article/details/123474312

版权

学习记录专栏收录该内容

103 篇文章 0 订阅

订阅专栏

102. 二叉树的层序遍历

在这里插入图片描述

class Solution:
    def levelOrder(self, root: TreeNode) -> List[List[int]]:
        if not root:
            return []
        res = []
        temps = [root]
        while temps:
            re = []
            temp = []
            for i in range(len(temps)):
                re.append(temps[i].val)
                if temps[i].left:
                    temp.append(temps[i].left)
                if temps[i].right:
                    temp.append(temps[i].right)
                
            temps = temp
            res.append(re)
        return res

199. 二叉树的右视图

在这里插入图片描述

class Solution:
    def rightSideView(self, root: TreeNode) -> List[int]:
        if not root:
            return []
        res = []
        temps = [root]
        while temps:
            res.append(temps[-1].val)
            temp = []
            for i in range(len(temps)):
                if temps[i].left:
                    temp.append(temps[i].left)
                if temps[i].right:
                    temp.append(temps[i].right)
            temps = temp
        return res

637. 二叉树的层平均值

在这里插入图片描述

# Definition for a binary tree node.
# class TreeNode:
#     def __init__(self, val=0, left=None, right=None):
#         self.val = val
#         self.left = left
#         self.right = right
class Solution:
    def averageOfLevels(self, root: Optional[TreeNode]) -> List[float]:
        if not root:
            return []
        res = []
        temps = [root]
        while temps:
            sum = 0
            l = len(temps)
            temp = []
            for i in range(len(temps)):
                sum += temps[i].val
                if temps[i].left:
                    temp.append(temps[i].left)
                if temps[i].right:
                    temp.append(temps[i].right)
            temps = temp
            res.append(sum/l)
        return res