二叉树的右视图

1. 题目描述

给定一棵二叉树，想象自己站在它的右侧，按照从顶部到底部的顺序，返回从右侧所能看到的节点值。

2. 思路

1. 层序遍历，取出最后一个元素即可

3. 实现

/**
 * Definition for a binary tree node.
 * type TreeNode struct {
 *     Val int
 *     Left *TreeNode
 *     Right *TreeNode
 * }
 */
func rightSideView(root *TreeNode) []int {
    if root == nil {
        return nil
    }

    var res []int
    currentLineLast := root
    var nextLineLast *TreeNode
    queue := []*TreeNode{root}
    for len(queue) > 0 {
        node := queue[0]
        queue = queue[1:]
        if node.Left != nil {
            queue = append(queue, node.Left)
            nextLineLast = node.Left
        }
        if node.Right != nil {
            queue = append(queue, node.Right)
            nextLineLast = node.Right
        }
        if node == currentLineLast {
            res = append(res, node.Val)
            currentLineLast = nextLineLast
        }
    }

    return res
}

4. 参考

• 199. 二叉树的右视图 - 力扣（LeetCode）