LeetCode笔记：515. Find Largest Value in Each Tree Row
LeetCode笔记：515. Find Largest Value in Each Tree Row
Cloudox
发布于 2021-11-23 16:27:50
发布于 2021-11-23 16:27:50
问题：
You need to find the largest value in each row of a binary tree. Example: Input:
Output: 1, 3, 9
大意：
你需要找到二叉树中每一行最大的值。 例子： 输入：
输出： 1, 3, 9
思路：
要找每一行最大的数，我们总归是要遍历二叉树的，遍历二叉树分为BFS和DFS，这里我们要找每行最大的数，那么我们就用BFS，一行行分析过去。
还记得我们在传送门：LeetCode笔记：102. Binary Tree Level Order Traversal中，是要求将二叉树一层层地输出出来。那么通过同样的方法，我们用利用队列的先进先出特性，依次保留每一行新读进来的节点。用一个变量记录当前行的总节点数，然后对每一行都去寻找最大的节点值是多少，记录在结果中就可以了。
代码（Java）：
/**
* Definition for a binary tree node.
* public class TreeNode {
* int val;
* TreeNode left;
* TreeNode right;
* TreeNode(int x) { val = x; }
* }
*/
public class Solution {
public List<Integer> largestValues(TreeNode root) {
Queue<TreeNode> queue = new LinkedList<TreeNode>();
List<Integer> res = new LinkedList<Integer>();
if (root == null) return res;
queue.offer(root);
while (!queue.isEmpty()) {
int levelNum = queue.size();
int temp = queue.peek().val;
if (queue.peek().left != null) queue.offer(queue.peek().left);
if (queue.peek().right != null) queue.offer(queue.peek().right);
queue.poll();
for (int i = 1; i < levelNum; i++) {
if (queue.peek().val > temp) temp = queue.peek().val;
if (queue.peek().left != null) queue.offer(queue.peek().left);
if (queue.peek().right != null) queue.offer(queue.peek().right);
queue.poll();
}
res.add(temp);
}
return res;
}
}他山之石：
除了用BFS，其实DFS也可以做，但是就需要有一个参数来记录当前节点所在的行，同时对于每次遇到的新节点，判断该节点值与已经记录的所在行的最大值之间的大小，如果更大就替换掉结果中记录的值，如果小一些那就略过。这个方法运行起来会比我的方法要快。
public class Solution {
public List<Integer> largestValues(TreeNode root) {
List<Integer> res = new ArrayList<Integer>();
helper(root, res, 0);
return res;
}
private void helper(TreeNode root, List<Integer> res, int d){
if(root == null){
return;
}
//expand list size
if(d == res.size()){
res.add(root.val);
}
else{
//or set value
res.set(d, Math.max(res.get(d), root.val));
}
helper(root.left, res, d+1);
helper(root.right, res, d+1);
}
}本文参与 腾讯云自媒体同步曝光计划，分享自作者个人站点/博客。
原始发表：2017/11/23 ，如有侵权请联系 cloudcommunity@tencent.com 删除
评论
登录后参与评论
推荐阅读
目录