#leetcode#515. Find Largest Element in Each Row

2037 단어 LeetCode

515. Find Largest Element in Each Row


My Submissions Back To Contest
User Accepted: 851 
User Tried: 878 
Total Accepted: 943 
Total Submissions: 2140 
Difficulty: Medium
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]

Discuss
------------------------------------------------------------------------------------------------------------------------------------------------------------------
아니면 BFS 문제입니다. 각 층마다 현재 층의 최대치를 기록하면 됩니다. 주의해야 할 것은 여기 input가null일 때 반환을 요구하는 것은 비어 있는 array입니다. 물론 실제 면접에서null로 돌아가는 것도 안 될 것이 없습니다.
/**
 * 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)    return new int[0];
        List list = new ArrayList<>();
        LinkedList queue = new LinkedList<>();
        queue.offer(root);
        while(!queue.isEmpty()){
            int tmpMax = Integer.MIN_VALUE;
            int size = queue.size();
            for(int i = 0; i < size; i++){
                TreeNode cur = queue.poll();
                tmpMax = Math.max(tmpMax, cur.val);
                if(cur.left != null)    queue.offer(cur.left);
                if(cur.right != null)   queue.offer(cur.right);
            }
            list.add(tmpMax);
        }
        int[] res = new int[list.size()];
        for(int j = 0; j < list.size(); j++){
            res[j] = list.get(j);
        }
        
        return res;
    }
}

좋은 웹페이지 즐겨찾기