zoukankan      html  css  js  c++  java
  • 637. Average of Levels in Binary Tree 二叉树的层次遍历再求均值

    [抄题]:

    Given a non-empty binary tree, return the average value of the nodes on each level in the form of an array.

    Example 1:

    Input:
        3
       / 
      9  20
        /  
       15   7
    Output: [3, 14.5, 11]
    Explanation:
    The average value of nodes on level 0 is 3,  on level 1 is 14.5, and on level 2 is 11. Hence return [3, 14.5, 11].

     [暴力解法]:

    时间分析:

    空间分析:

    [奇葩输出条件]:

    [奇葩corner case]:

    [思维问题]:

    BFS写不熟

    [一句话思路]:

    (3先生)先加头、先判Empty、先取长度

    [输入量]:空: 正常情况:特大:特小:程序里处理到的特殊情况:异常情况(不合法不合理的输入):

    [画图]:

    [一刷]:

    1. 没概念:BFS中的for循环是针对当前这一层的操作,add的元素都属于下一层
    2. 二叉树层遍历,q中存的是节点,记住就行了

    [二刷]:

    [三刷]:

    [四刷]:

    [五刷]:

      [五分钟肉眼debug的结果]:

    [总结]:

    BFS里既然取了长度,就用在for循环中

    [复杂度]:Time complexity: O(n) Space complexity: O(n)

    所有点放进去一次,拿出来一次,最终还是n

    [英文数据结构或算法,为什么不用别的数据结构或算法]:

    [关键模板化代码]:

    q.offer(root);
            //isEmpty()
            while (! q.isEmpty()) {
                //get length
                int n = q.size();

    [其他解法]:

    [Follow Up]:

    [LC给出的题目变变变]:

     [代码风格] :

    /**
     * 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) {
            //corner case
            List<Double> ans = new ArrayList<Double>();
            Queue<TreeNode> q = new LinkedList<TreeNode>();
        
            if (root == null) {
                return ans;
            }
            //bfs
            //offer root
            q.offer(root);
            //isEmpty()
            while (! q.isEmpty()) {
                //get length
                int n = q.size();
                double sum = 0.0;
                for (int i = 0; i < n; i++) {
                    TreeNode node = q.poll();
                    sum += node.val;
                    if (node.left != null) {
                        q.offer(node.left);
                    }
                    if (node.right != null) {
                        q.offer(node.right);
                    }
                }
                ans.add(sum / n);
            }
            //return
            return ans;
        }
    }
    View Code
  • 相关阅读:
    Anagram
    HDU 1205 吃糖果(鸽巢原理)
    Codeforces 1243D 0-1 MST(补图的连通图数量)
    Codeforces 1243C Tile Painting(素数)
    Codeforces 1243B2 Character Swap (Hard Version)
    Codeforces 1243B1 Character Swap (Easy Version)
    Codeforces 1243A Maximum Square
    Codeforces 1272E Nearest Opposite Parity(BFS)
    Codeforces 1272D Remove One Element
    Codeforces 1272C Yet Another Broken Keyboard
  • 原文地址:https://www.cnblogs.com/immiao0319/p/8587624.html
Copyright © 2011-2022 走看看