给定一个非空二叉树, 返回一个由每层节点平均值组成的数组。
(节点值的范围在32位有符号整数范围内)
输入:
[3,9,20,15,7]
输出:
[3.00000,14.50000,11.00000]
class Solution {
double[] a=new double[1000];
int[] b=new int[1000];
int count=0;
public List<Double> averageOfLevels(TreeNode root) {
List<Double> list=new ArrayList<Double>();
dfs(root,0);
for(int i=0;i<=count;i++){
list.add(a[i]*1.0/b[i]);
}
return list;
}
public void dfs(TreeNode r,int k){
if(r.left==null&&r.right==null){
a[k]+=r.val;
b[k]++;
count=k>count?k:count;
}else{
a[k]+=r.val;
b[k]++;
if(r.right!=null){
dfs(r.right,k+1);
}
if(r.left!=null){
dfs(r.left,k+1);
}
}
}
}
//leetcode提交执行用时内存消耗
如果你的提交不尽人意,建议多提交几次—#滑稽