来自北大算法课的Leetcode题解：637. 二叉树的层平均值

最新推荐文章于 2024-08-31 20:59:22 发布

doubleZ0108

最新推荐文章于 2024-08-31 20:59:22 发布

阅读量95

点赞数

分类专栏：来自北大算法课的Leetcode题解文章标签：算法 leetcode

本文链接：https://blog.csdn.net/double_ZZZ/article/details/128864709

版权

来自北大算法课的Leetcode题解专栏收录该内容

247 篇文章 5 订阅

订阅专栏

代码仓库：Github | Leetcode solutions @doubleZ0108 from Peking University.

解法1(JS T15.33% S9%)：还是正常树的层序遍历，现将每个节点的值都保存在对应层级的数组中，然后再用数组的map()方法将每一层的数组求平均值
- 注意添加左右节点的时候是对node来说的，不要递归写多了都成root的左右节点反复加了

var averageOfLevels = function(root) {
    var res = [];
    var queue = [[root, 1]];
    while (queue.length) {
        var data = queue.shift();
        var node=data[0], level=data[1];

        if (res.length < level) {
            res.push([node.val]);
        } else {
            res[level-1].push(node.val);
        }

        if (node.left) {
            queue.push([node.left, level+1]);
        }
        if (node.right) {
            queue.push([node.right, level+1]);
        }
    }

    return [...res.map(x => _.sum(x)/x.length)];
};