題目

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

題解

1、參考“二叉樹按層打印”寫的解法

詳細思路參考：左神算法：二叉樹的按層打印與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; // 當前行的最右節點TreeNode nLast = null; // 下一行的最右節點queue.offer(root);double sum = 0;double cnt = 0;while (!queue.isEmpty()) { // 每一層都做從左到右寬度優先遍歷root = queue.poll();sum += root.val;cnt++;if (root.left != null) {queue.offer(root.left);nLast = root.left; // 讓nLast一直跟蹤記錄寬度優先隊列中的最新加入的節點即可，因為最新加入隊列的節點一定是目前已經發現的下一行的最右節點}if (root.right != null) {queue.offer(root.right);nLast = root.right;}if (root == last) { // 如果發現遍歷到的節點等于last，則說明應該換行list.add(sum / cnt);sum = 0;cnt = 0;last = nLast; // 當前行累加完時，nLast一定是下一行所有節點中的最右節點，只要令last=nLast，就可以繼續下一行的累加過程}}return list;} }

2、評論區簡潔解法

/*** 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;} }

總結

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

如果覺得生活随笔網站內容還不錯，歡迎將生活随笔推薦給好友。