[Leetcode] 637. 二叉樹的層平均值 java
阿新 • • 發佈:2018-12-16
給定一個非空二叉樹, 返回一個由每層節點平均值組成的陣列.
示例 1:
輸入: 3 / \ 9 20 / \ 15 7 輸出: [3, 14.5, 11] 解釋: 第0層的平均值是 3, 第1層是 14.5, 第2層是 11. 因此返回 [3, 14.5, 11].
注意:
- 節點值的範圍在32位有符號整數範圍內。
/** * 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) { List<Double> result=new ArrayList(); if(root==null) return result; List<TreeNode> list=new ArrayList(); list.add(root); while(!list.isEmpty()){ List<Integer> curList=new ArrayList(); List<TreeNode> nextList=new ArrayList(); int count=0; double sum=0; for(TreeNode cur:list){ sum+=cur.val; count++; if(cur.left!=null) nextList.add(cur.left); if(cur.right!=null) nextList.add(cur.right); } double average=sum/count; result.add(average); list=nextList; } return result; } }