199. 二叉树的右视图

image.png

1广度优先遍历二叉树,右边所能看到的肯定是当前层的最后一个节点,所以这个就是求每层的最后一个节点数
2 处理好出队和入队的先后关系, 出队是把当前队列的数据都出了。但是在出队之前得把下一层数据都入队了。

  1. package main
  2. import "fmt"
  3. type TreeNode struct {
  4. Val int
  5. Left *TreeNode
  6. Right *TreeNode
  7. }
  8. func rightSideView(root *TreeNode) []int {
  9. if root == nil {
  10. return nil
  11. }
  12. queue := make([]*TreeNode, 0)
  13. queue = append(queue, root)
  14. var res []int
  15. for len(queue) > 0 {
  16. l := len(queue)
  17. node := queue[l-1]
  18. res = append(res, node.Val)
  19. for i := range queue {
  20. currNode := queue[i]
  21. if currNode.Left != nil {
  22. queue = append(queue, currNode.Left)
  23. }
  24. if currNode.Right != nil {
  25. queue = append(queue, currNode.Right)
  26. }
  27. }
  28. queue = queue[l:]
  29. }
  30. return res
  31. }
  32. func main() {
  33. one := &TreeNode{Val: 1}
  34. two := &TreeNode{Val: 2}
  35. one.Left = two
  36. three := &TreeNode{Val: 3}
  37. one.Right = three
  38. //five :=&TreeNode{Val: 5}
  39. //two.Right = five
  40. four := &TreeNode{Val: 4}
  41. two.Left = four
  42. fmt.Println(rightSideView(one))
  43. //fmt.Println(rightSideView(one))
  44. }