題目

https://leetcode-cn.com/problems/average-of-levels-in-binary-tree/

題解

1、參考“二叉樹(shù)按層打印”寫(xiě)的解法

詳細(xì)思路參考：左神算法：二叉樹(shù)的按層打印與ZigZag打印（Java版）

import java.util.ArrayList; import java.util.LinkedList; import java.util.List; import java.util.Queue;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) {ArrayList<Double> list = new ArrayList<>();Queue<TreeNode> queue = new LinkedList<>();TreeNode last = root; // 當(dāng)前行的最右節(jié)點(diǎn)TreeNode nLast = null; // 下一行的最右節(jié)點(diǎn)queue.offer(root);double sum = 0;double cnt = 0;while (!queue.isEmpty()) { // 每一層都做從左到右寬度優(yōu)先遍歷root = queue.poll();sum += root.val;cnt++;if (root.left != null) {queue.offer(root.left);nLast = root.left; // 讓nLast一直跟蹤記錄寬度優(yōu)先隊(duì)列中的最新加入的節(jié)點(diǎn)即可，因?yàn)樽钚录尤腙?duì)列的節(jié)點(diǎn)一定是目前已經(jīng)發(fā)現(xiàn)的下一行的最右節(jié)點(diǎn)}if (root.right != null) {queue.offer(root.right);nLast = root.right;}if (root == last) { // 如果發(fā)現(xiàn)遍歷到的節(jié)點(diǎn)等于last，則說(shuō)明應(yīng)該換行list.add(sum / cnt);sum = 0;cnt = 0;last = nLast; // 當(dāng)前行累加完時(shí)，nLast一定是下一行所有節(jié)點(diǎn)中的最右節(jié)點(diǎn)，只要令last=nLast，就可以繼續(xù)下一行的累加過(guò)程}}return list;} }

2、評(píng)論區(qū)簡(jiǎn)潔解法

/*** 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> res = new ArrayList<>();if (root == null) return res;Queue<TreeNode> list = new LinkedList<>();list.add(root);while (list.size() != 0){int len = list.size();double sum = 0;for (int i = 0; i < len; i++){TreeNode node = list.poll();sum += node.val;if (node.left != null) list.add(node.left);if (node.right != null) list.add(node.right);}res.add(sum/len);}return res;} }

總結(jié)

以上是生活随笔為你收集整理的leetcode 637. Average of Levels in Binary Tree | 637. 二叉树的层平均值（Java）的全部?jī)?nèi)容，希望文章能夠幫你解決所遇到的問(wèn)題。

如果覺(jué)得生活随笔網(wǎng)站內(nèi)容還不錯(cuò)，歡迎將生活随笔推薦給好友。