515. Find Largest Value in Each Tree Row

You need to find the largest value in each row of a binary tree.

Example:

Input: 

          1
         / \
        3   2
       / \   \  
      5   3   9 

Output: [1, 3, 9]
题目含义:从上到下找到每行的最大值
方法一:DFS
参数d表示层数的索引,第一行对应的d为0
复制代码
 1     private void largestValue(TreeNode root, List<Integer> res, int d){
 2         if(root == null){
 3             return;
 4         }
 5         //expand list size
 6         if(d == res.size()){
 7             res.add(root.val); //在最下面添加新的行
 8         }
 9         else{
10             //or set value
11             res.set(d, Math.max(res.get(d), root.val)); //当前节点值处于第(d+1)层,如果当前值大于res中保存的值,则替换成最大值
12         }
13         largestValue(root.left, res, d+1);
14         largestValue(root.right, res, d+1);
15     }
16     
17     
18     public List<Integer> largestValues(TreeNode root) {
19         List<Integer> res = new ArrayList<Integer>();
20         largestValue(root, res, 0);
21         return res;     
22     }
复制代码

 

方法二:

复制代码
 1     public List<Integer> largestValues(TreeNode root) {
 2         Queue<TreeNode> queue = new LinkedList<TreeNode>();
 3         List<Integer> res = new ArrayList<Integer>();
 4         if (root == null) return res;
 5         queue.add(root);
 6         while (!queue.isEmpty()) {
 7             int largestElement = Integer.MIN_VALUE;
 8             int queueSize = queue.size();
 9             for (int i=0;i<queueSize;i++) {
10                 TreeNode cur = queue.poll();
11                 largestElement = Math.max(cur.val, largestElement);
12                 if (cur.left != null) queue.add(cur.left);
13                 if (cur.right != null) queue.add(cur.right);
14             }
15             res.add(largestElement);
16         }
17         return res;
18     }
复制代码

 


posted @   daniel456  阅读(211)  评论(0编辑  收藏  举报
编辑推荐:
· AI与.NET技术实操系列:基于图像分类模型对图像进行分类
· go语言实现终端里的倒计时
· 如何编写易于单元测试的代码
· 10年+ .NET Coder 心语,封装的思维:从隐藏、稳定开始理解其本质意义
· .NET Core 中如何实现缓存的预热?
阅读排行:
· 25岁的心里话
· 闲置电脑爆改个人服务器(超详细) #公网映射 #Vmware虚拟网络编辑器
· 基于 Docker 搭建 FRP 内网穿透开源项目(很简单哒)
· 零经验选手,Compose 一天开发一款小游戏!
· 一起来玩mcp_server_sqlite,让AI帮你做增删改查!!
点击右上角即可分享
微信分享提示