二叉树3. 层次遍历之二:3道变形题目
Posted 纵横千里,捭阖四方
tags:
篇首语:本文由小常识网(cha138.com)小编为大家整理,主要介绍了二叉树3. 层次遍历之二:3道变形题目相关的知识,希望对你有一定的参考价值。
我们继续看几个典型的层次遍历的题目
515.在每个树行中找最大值
637.二叉树的层平均值
199.二叉树的右视图
这三道题,515和637的思路基本一致,就是先要确定每层的元素,之后再求最大、最小、平均,都不在话下。
1.LeetCode 515. 在每个树行中找最大值
给定一棵二叉树的根节点 root ,请找出该二叉树中每一层的最大值。
这里其实就是在得到一层之后使用一个变量来记录当前得到的最大值:
class Solution {
public List<Integer> largestValues(TreeNode root) {
List<Integer> res = new ArrayList<>();
Deque<TreeNode> deque = new ArrayDeque<>();
if (root != null) {
deque.addLast(root);
}
while (!deque.isEmpty()) {
int size = deque.size();
int levelMaxNum = Integer.MIN_VALUE;
for (int i = 0; i < size; i++) {
TreeNode node = deque.poll();
levelMaxNum = Math.max(node.val,levelMaxNum);
if (node.left != null) deque.addLast(node.left);
if (node.right != null) deque.addLast(node.right);
}
res.add(levelMaxNum);
}
return res;
}
}
2.LeetCode 637. 在每个树行中找平均值
要求给定一个非空二叉树, 返回一个由每层节点平均值组成的数组。
例如:
这个题和前面的几个一样,只不过是每层都先将元素保存下来,最后求平均就行了:
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;
}
}
3.LeetCode 199.二叉树的右视图
题目要求是:给定一个二叉树的 根节点 root,想象自己站在它的右侧,按照从顶部到底部的顺序,返回从右侧所能看到的节点值。
例如:
这个题目出现频率还挺高的,如果没有提前思考过,面试现场可能会想不到怎么做。
其实也很简单那,利用 BFS 进行层次遍历,记录下每层的最后一个元素。
class Solution {
public List<Integer> rightSideView(TreeNode root) {
List<Integer> res = new ArrayList<>();
if (root == null) {
return res;
}
Queue<TreeNode> queue = new LinkedList<>();
queue.offer(root);
while (!queue.isEmpty()) {
int size = queue.size();
for (int i = 0; i < size; i++) {
TreeNode node = queue.poll();
if (node.left != null) {
queue.offer(node.left);
}
if (node.right != null) {
queue.offer(node.right);
}
if (i == size - 1) { //将当前层的最后一个节点放入结果列表
res.add(node.val);
}
}
}
return res;
}
}
是不是很简单,轻轻松松搞定3道题,而这三个题本质都是层次遍历的调整。
以上是关于二叉树3. 层次遍历之二:3道变形题目的主要内容,如果未能解决你的问题,请参考以下文章