二叉树层序遍历-求每层平均值

最新推荐文章于 2024-11-17 12:24:42 发布

凌晨三点的北京

最新推荐文章于 2024-11-17 12:24:42 发布

阅读量151

点赞数

文章标签： java javascript 前端

本文链接：https://blog.csdn.net/weixin_38687303/article/details/133913713

版权

使用队列存储当前层级的节点个数

fun averageOfLevels(root: TreeNode?): DoubleArray {
        var queue = LinkedList<TreeNode>()
        val sums = arrayListOf<Double>()
        if (null == root) return doubleArrayOf()
        queue.add(root)
        var size = 1
        while (queue.isNotEmpty()) {
            var tempSum = 0.0
            val tempCount = size
            var nextLCount = 0
            while (size > 0) {
                val node = queue.poll()
                tempSum += node.`val`
                if (null != node.left) {
                    queue.add(node.left)
                    nextLCount++
                }
                if (null != node.right) {
                    queue.add(node.right)
                    nextLCount++
                }
                size--
            }
            sums.add(tempSum / tempCount)
            size = nextLCount
        }
        val doubleArray = DoubleArray(sums.size)
        sums.forEachIndexed { index, d -> 
            doubleArray.set(index,d)
        }
        return doubleArray
    }