【二叉树】二叉树的层平均值

最新推荐文章于 2024-06-04 10:30:28 发布

豪冷啊

最新推荐文章于 2024-06-04 10:30:28 发布

阅读量861

点赞数 1

文章标签：算法 leetcode swift ios objective-c

本文链接：https://blog.csdn.net/xjh093/article/details/122358859

版权

二叉树层序遍历节点平均值 Swift 数据结构

关键词由CSDN通过智能技术生成

0x00 题目

给定一个非空二叉树
返回一个由每层节点平均值组成的数组

输入：

输出：[3, 14.5, 11]

解释：
第 0 层的平均值是 3 , 第 1 层是 14.5 , 第 2 层是 11
因此返回 [3, 14.5, 11]

0x01 思路

要求出每层的 平均值
那就得先求每层的 和
以及每层的节点 总个数

只能使用 层序 遍历的方式了

0x02 解法

语言：Swift

树节点：TreeNode

public class TreeNode {
    public var val: Int
    public var left: TreeNode?
    public var right: TreeNode?
    public init() { self.val = 0; self.left = nil; self.right = nil; }
    public init(_ val: Int) { self.val = val; self.left = nil; self.right = nil; }
    public init(_ val: Int, _ left: TreeNode?, _ right: TreeNode?) {
        self.val = val
        self.left = left
        self.right = right
    }
}

解法：

func averageOfLevels(_ root: TreeNode?) -> [Double] {
    guard let root = root else { return [] }
    
    // 用来存储每层的节点
    var queue: [TreeNode] = []
    queue.append(root)
    
    // 用来存储平均值
    var result: [Double] = []

    while !queue.isEmpty {
        // 用来存储下一层的所有节点
        var tmp: [TreeNode] = []
        
        // 当前层的 和
        var sum = 0
        
        // 当前层的 个数
        let count = Double(queue.count)
        
        // 获取下一层的所有节点
        while !queue.isEmpty {
            let node = queue.removeFirst()
            // 同时计算和
            sum += node.val

            if let left = node.left {
                tmp.append(left)
            }
            if let right = node.right {
                tmp.append(right)
            }
        }
        
        // 切换到下一层
        queue = tmp
        
        // 当前层的平均值
        result.append(Double(sum)/count)
    }
    
    return result
}