[LeetCode] Binary Tree Right Side View 문제 풀이 보고서

[제목]
Given a binary tree, imagine yourself standing on the right side of it, return the values of the nodes you can see ordered from top to bottom.
For example: Given the following binary tree,
   1            <---
 /   \
2     3         <---
 \     \
  5     4       <---

You should return  [1, 3, 4] . [해석]
제목: 두 갈래 나무를 정하고 오른쪽에서 이 두 갈래 나무가 보이는 노드 서열을 되돌려줍니다.
사고방식: 차원 역법.각 층의 마지막 노드를 두루 훑어보았을 때, 그것을 결과에 집중시켰다.
[Java 코드]
public class Solution {
    public List<Integer> rightSideView(TreeNode root) {
        List<Integer> ans = new ArrayList<Integer>();
        if (root == null) return ans;
        
        LinkedList<TreeNode> queue = new LinkedList<TreeNode>();
        queue.add(root);
        queue.add(null);
        
        while (!queue.isEmpty()) {
            TreeNode node = queue.pollFirst();
            
            if (node == null) {
                if (queue.isEmpty()) {
                    break;
                } else {
                    queue.add(null);
                }
            } else {
                // add the rightest to the answer
                if (queue.peek() == null) {
                    ans.add(node.val);
                }
                
                if (node.left != null) {
                    queue.add(node.left);
                }
                if (node.right != null) {
                    queue.add(node.right);
                }
            }
        }
        
        return ans;
    }
}

좋은 웹페이지 즐겨찾기