CodingTest/LeetCode

[LeetCode] 199. Binary Tree Right Side View, Medium

뜸부깅 2025. 4. 25. 14:05
반응형

1. 문제

  • 이진 트리가 주어질 때, 오른쪽에서 바라볼 때 보이는 노드의 값을 담은 배열을 반환하라.

2. 해결

/**
 * Definition for a binary tree node.
 * class TreeNode {
 *     val: number
 *     left: TreeNode | null
 *     right: TreeNode | null
 *     constructor(val?: number, left?: TreeNode | null, right?: TreeNode | null) {
 *         this.val = (val===undefined ? 0 : val)
 *         this.left = (left===undefined ? null : left)
 *         this.right = (right===undefined ? null : right)
 *     }
 * }
 */

function rightSideView(root: TreeNode | null): number[] {
    if(!root) return [];
    const queue = [root]
    const ans = [];

    while(queue.length) {
        const size = queue.length;
        
        ans.push(queue[size-1].val)
        for(let i = 0; i < size; i++) {
            const node = queue.shift();
            if(node.left) queue.push(node.left);
            if(node.right) queue.push(node.right);
        }
        
    }

    return ans;
};
  • 큐를 이용한 BFS 탐색을 하고, 그 과정에서 담기는 레벨 별 노드의 마지막 노드가 오른쪽에서 바라볼 때 보이는 노드다.