每日一题算法：2020年9月12日[二叉树的层平均值] averageOfLevels

2020年9月12日二叉树的层平均值 averageOfLevels

[外链图片转存失败,源站可能有防盗链机制,建议将图片保存下来直接上传(img-EI7Gm6G0-1599919552885)(C:%5CUsers%5CAdmin%5CAppData%5CRoaming%5CTypora%5Ctypora-user-images%5Cimage-20200912213744908.png)]

/**
 * Definition for a binary tree node.
 * public class TreeNode {
 *     int val;
 *     TreeNode left;
 *     TreeNode right;
 *     TreeNode(int x) { val = x; }
 * }
 */
class Solution {
    public List<Double> averageOfLevels(TreeNode root) {

    }
}

解题思路：

这种树状的问题，一般都可以尝试能否使用广度优先算法和深度优先算法，这道题就比较适合使用广度优先算法的变种。

1，我们构建一个链表、队列

2，将根节点装进队列中

3，取出队列中所有的节点，并且将所有节点的子节点装入队列中

4，重复过程3，直到队列为空，即没有子节点了。

代码实现：

[外链图片转存失败,源站可能有防盗链机制,建议将图片保存下来直接上传(img-n5Gwl162-1599919552890)(C:%5CUsers%5CAdmin%5CAppData%5CRoaming%5CTypora%5Ctypora-user-images%5Cimage-20200912220515153.png)]

    public List<Double> averageOfLevels(TreeNode root) {
        List<Double> res=new ArrayList<>();
        
        LinkedList<TreeNode> list=new LinkedList<>();
        
        if (root!=null)
            list.add(root);
        else 
            return res;
        
        while (list.size()!=0){
            int len=list.size();
            
            Double sum=0.0;
            for (int i=0;i<len;i++){
                TreeNode first = list.getFirst();
                
                sum+=first.val;
                if (first.right!=null)
                    list.add(first.right);
                if (first.left!=null)
                    list.add(first.left);
                
                list.removeFirst();
            }
            res.add(sum/(len*1.0));
            
        }
        
        return res;
    }