LeetCode赛题515----Find Largest Element in Each Row

问题描述

You need to find the largest element in each row of a Binary Tree.

Example:

Input: 

      1
     / \
    2   3
   / \   \  
  5   3   9 

Output: [1, 3, 9]

算法分析

使用两个队列，逐层遍历二叉树的各个节点，每个队列中的节点都是同一层的节点，在遍历一层时，找出该层最大的节点值加入ArrayList中。

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 int[] findValueMostElement(TreeNode root) {
        if(root==null){
    		int[] tmp=new int[0];
    		return tmp;
    	}
        ArrayList<Integer>arr=new ArrayList<>();
        Queue<TreeNode>que1=new LinkedList<>();
        Queue<TreeNode>que2=new LinkedList<>();
        que1.add(root);
        int max;
        while(!que1.isEmpty()||!que2.isEmpty()){
        	max=Integer.MIN_VALUE;
        	if(!que1.isEmpty()){
        		while(!que1.isEmpty()){
        			TreeNode node=que1.poll();
        			if(node.val>max){	//找到该层的最大值
        				max=node.val;
        			}
        			if(node.left!=null){
        				que2.add(node.left);
        			}
        			if(node.right!=null){
        				que2.add(node.right);
        			}
        		}
        		arr.add(max);
        	}
        	else if(!que2.isEmpty()){
        		while(!que2.isEmpty()){
        			TreeNode node=que2.poll();
        			if(node.val>max){
        				max=node.val;
        			}
        			if(node.left!=null){
        				que1.add(node.left);
        			}
        			if(node.right!=null){
        				que1.add(node.right);
        			}
        		}
        		arr.add(max);
        	}
        }
        int [] ans=new int[arr.size()];
        for(int i=0;i<ans.length;i++){
        	ans[i]=arr.get(i);
        }
        return ans;
    }
}