637:二叉树的层平均值

This commit is contained in:
轩辕龙儿 2022-03-28 17:29:24 +08:00
parent 0f5c11beba
commit 443a60fa43
2 changed files with 128 additions and 0 deletions

View File

@ -0,0 +1,93 @@
//给定一个非空二叉树的根节点 root , 以数组的形式返回每一层节点的平均值与实际答案相差 10 以内的答案可以被接受
//
//
//
// 示例 1
//
//
//
//
//输入root = [3,9,20,null,null,15,7]
//输出[3.00000,14.50000,11.00000]
//解释 0 层的平均值为 3, 1 层的平均值为 14.5, 2 层的平均值为 11
//因此返回 [3, 14.5, 11]
//
//
// 示例 2:
//
//
//
//
//输入root = [3,9,20,15,7]
//输出[3.00000,14.50000,11.00000]
//
//
//
//
// 提示
//
//
//
//
// 树中节点数量在 [1, 10] 范围内
// -2³¹ <= Node.val <= 2³¹ - 1
//
// Related Topics 深度优先搜索 广度优先搜索 二叉树 👍 319 👎 0
package leetcode.editor.cn;
import com.code.leet.entiy.TreeNode;
import java.util.ArrayList;
import java.util.LinkedList;
import java.util.List;
import java.util.Queue;
//637:二叉树的层平均值
public class AverageOfLevelsInBinaryTree{
public static void main(String[] args) {
Solution solution = new AverageOfLevelsInBinaryTree().new Solution();
// TO TEST
}
//leetcode submit region begin(Prohibit modification and deletion)
/**
* Definition for a binary tree node.
* public class TreeNode {
* int val;
* TreeNode left;
* TreeNode right;
* TreeNode() {}
* TreeNode(int val) { this.val = val; }
* TreeNode(int val, TreeNode left, TreeNode right) {
* this.val = val;
* this.left = left;
* this.right = right;
* }
* }
*/
class Solution {
public List<Double> averageOfLevels(TreeNode root) {
Queue<TreeNode> queue = new LinkedList<>();
queue.add(root);
List<Double> result = new ArrayList<>();
while (!queue.isEmpty()){
int size = queue.size();
double sum = 0f;
for (int i = 0; i < size; i++) {
TreeNode node = queue.poll();
if(node.left!=null){
queue.add(node.left);
}
if(node.right!=null){
queue.add(node.right);
}
sum+= node.val;
}
result.add(sum/size);
}
return result;
}
}
//leetcode submit region end(Prohibit modification and deletion)
}

View File

@ -0,0 +1,35 @@
<p>给定一个非空二叉树的根节点<meta charset="UTF-8" />&nbsp;<code>root</code>&nbsp;, 以数组的形式返回每一层节点的平均值。与实际答案相差&nbsp;<code>10<sup>-5</sup></code> 以内的答案可以被接受。</p>
<p>&nbsp;</p>
<p><strong>示例 1</strong></p>
<p><img alt="" src="https://assets.leetcode.com/uploads/2021/03/09/avg1-tree.jpg" /></p>
<pre>
<strong>输入:</strong>root = [3,9,20,null,null,15,7]
<strong>输出:</strong>[3.00000,14.50000,11.00000]
<strong>解释:</strong>第 0 层的平均值为 3,第 1 层的平均值为 14.5,第 2 层的平均值为 11 。
因此返回 [3, 14.5, 11] 。
</pre>
<p><strong>示例 2:</strong></p>
<p><img alt="" src="https://assets.leetcode.com/uploads/2021/03/09/avg2-tree.jpg" /></p>
<pre>
<strong>输入:</strong>root = [3,9,20,15,7]
<strong>输出:</strong>[3.00000,14.50000,11.00000]
</pre>
<p>&nbsp;</p>
<p><strong>提示:</strong></p>
<p><meta charset="UTF-8" /></p>
<ul>
<li>树中节点数量在&nbsp;<code>[1, 10<sup>4</sup>]</code> 范围内</li>
<li><code>-2<sup>31</sup>&nbsp;&lt;= Node.val &lt;= 2<sup>31</sup>&nbsp;- 1</code></li>
</ul>
<div><div>Related Topics</div><div><li></li><li>深度优先搜索</li><li>广度优先搜索</li><li>二叉树</li></div></div><br><div><li>👍 319</li><li>👎 0</li></div>