给定一个二叉树的 根节点 root,想象自己站在它的右侧,按照从顶部到底部的顺序,返回从右侧所能看到的节点值。
// 层序遍历,输出每层最右边的节点
func rightSideView(root *TreeNode) []int {
if root == nil {
return nil
}
res := [][]int{}
result := []int{}
queue := []*TreeNode{root}
for len(queue) > 0 {
length := len(queue)
var ans = []int{}
for i := 0; i < length; i++ {
temp := queue[0]
queue = queue[1:]
ans = append(ans, temp.Val)
if temp.Left != nil {
queue = append(queue, temp.Left)
}
if temp.Right != nil {
queue = append(queue, temp.Right)
}
}
res = append(res, ans)
}
for _, v := range res {
length := len(v)
result = append(result, v[length-1])
}
return result
}