LeetCode简单题之二叉树的层平均值
生活随笔
收集整理的這篇文章主要介紹了
LeetCode简单题之二叉树的层平均值
小編覺得挺不錯的,現在分享給大家,幫大家做個參考.
題目
給定一個非空二叉樹的根節點 root , 以數組的形式返回每一層節點的平均值。與實際答案相差 10-5 以內的答案可以被接受。
示例 1:
輸入:root = [3,9,20,null,null,15,7]
輸出:[3.00000,14.50000,11.00000]
解釋:第 0 層的平均值為 3,第 1 層的平均值為 14.5,第 2 層的平均值為 11 。
因此返回 [3, 14.5, 11] 。
示例 2:
輸入:root = [3,9,20,15,7]
輸出:[3.00000,14.50000,11.00000]
提示:
樹中節點數量在 [1, 104] 范圍內
-2^31 <= Node.val <= 2 ^31 - 1
來源:力扣(LeetCode)
解題思路
??一個非常直觀的思路就是層序遍歷的變種。修改層序遍歷讓隊列每次都清空之后再進行入隊操作,這也可以有效的計算一層有多少結點以及每層的總和。
# 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]:Q=deque()Q.append(root)ans=[]temp=[]while Q:q=Q.popleft()temp.append(q)if len(Q)==0:ans.append(sum([j.val for j in temp])/len(temp))for i in temp:if i.left:Q.append(i.left)if i.right:Q.append(i.right)temp.clear()return ans
總結
以上是生活随笔為你收集整理的LeetCode简单题之二叉树的层平均值的全部內容,希望文章能夠幫你解決所遇到的問題。
- 上一篇: LeetCode简单题之数组异或操作
- 下一篇: LeetCode简单题之整理字符串