637二叉树的层平均值

二叉树一来就是一个一脸懵逼，数据结构没咋认真学过，完全不太会。

题目描述

给定一个非空二叉树的根节点 root , 以数组的形式返回每一层节点的平均值。与实际答案相差 10-5 以内的答案可以被接受。

示例

输入：root = [3,9,20,15,7]
输出：[3.00000,14.50000,11.00000]

思路

深度优先遍历二叉树，记录每一层的节点数以及节点数值之和，最后求个平均，看起来很简单，用递归

代码

深度优先：

public void dfs(TreeNode treeNode, int level, List<Integer> counts, List<Integer> sums) {
        
        if (treeNode == null) {
            return;
        }
        if (level < sums.size()) {
            counts.set(level, counts.get(level) + 1);
            sums.set(level, sums.get(level) + treeNode.val);
        } else {
            counts.add(1);
            sums.add(treeNode.val);
        }
        dfs(treeNode.left, level + 1, counts, sums);
        dfs(treeNode.right, level + 1, counts, sums);
        
    }

求平均：

public List<Double> averageOfLevels(TreeNode root) {
        List<Integer> counts = new ArrayList<>();
        List<Integer> sums = new ArrayList<>();
        dfs(root,0,counts,sums);
        List<Double> ave = new ArrayList<>();
        for (int i = 0; i < sums.size(); i++) {
            ave.add((double) (sums.get(i) / counts.get(i)));
        }
        return ave;
    }

思路很简单，但是对二叉树这个数据结构不熟悉，一开始不知道咋操作，搜了一下深度优先的实现，才写出来。