Leetcode刷题(第102题)——二叉树的层序遍历

x33g5p2x  于2022-03-09 转载在 其他  
字(0.8k)|赞(0)|评价(0)|浏览(375)

一、题目

  1. 给你二叉树的根节点 root ,返回其节点值的 层序遍历 (即逐层地,从左到右访问所有节点)。

二、示例

  1. 输入:root = [3,9,20,null,null,15,7]
  2. 输出:[[3],[9,20],[15,7]]
  1. 输入:root = [1]
  2. 输出:[[1]]
  1. 输入:root = []
  2. 输出:[]

三、思路
二叉树的层序遍历采用的方法是:使用到的数据结构是队列。只不过在本题需要加上层号,方便分组。
四、代码

  1. /**
  2. * Definition for a binary tree node.
  3. * function TreeNode(val, left, right) {
  4. * this.val = (val===undefined ? 0 : val)
  5. * this.left = (left===undefined ? null : left)
  6. * this.right = (right===undefined ? null : right)
  7. * }
  8. */
  9. /**
  10. * @param {TreeNode} root
  11. * @return {number[][]}
  12. */
  13. var levelOrder = function(root) {
  14. let result = []
  15. let arr = []
  16. if(!root) return result
  17. result.push([0, root])
  18. while(result.length) {
  19. let curArr = result.shift()
  20. let index = curArr[0]
  21. let val = curArr[1]
  22. if(arr[index]) {
  23. arr[index].push(val.val)
  24. }else {
  25. let n = val.val
  26. arr[index] = [n]
  27. }
  28. if(val.left) result.push([index + 1, val.left])
  29. if(val.right) result.push([index + 1, val.right])
  30. }
  31. return arr
  32. };

五、总结

相关文章