[路飞]_leetcode-103-二叉树的锯齿形层序遍历

[题目地址] [B站地址]

给定一个二叉树,返回其节点值的锯齿形层序遍历。(即先从左往右,再从右往左进行下一层遍历,以此类推,层与层之间交替进行)。

例如:
给定二叉树 [3,9,20,null,null,15,7],

    3
   / \
  9  20
    /  \
   15   7
复制代码

返回锯齿形层序遍历如下:

[
  [3],
  [20,9],
  [15,7]
]
复制代码

本题只是在二叉树层序遍历的基础上加了一个交替翻转遍历方向的逻辑 接下来,我们一起回顾下二叉树的层序遍历过程:

  1. 创建结果数组
  2. 在普通二叉树的递归遍历过程中传入对应节点的深度,根据当前节点的深度,将其节点值插入结果数组对应子数组中
  3. 返回结果数组即可

本题要求完成锯齿形层序遍历,只需要在偶数行的时候,从左向右遍历,奇数行的时候从右向左遍历。

这里我们没有必要真正的切换遍历顺序,只需要在节点值插入结果数组对应子数组时,判断当前是深度是奇数还是偶数
如果为偶数,采用push 方法将节点值插入子数组末尾
如果为奇数,采用unshift 方法将结果值插入子数组开头

代码如下:

var zigzagLevelOrder = function(root) {
  // 初始化结果数组
  const res = [];
  // 前序遍历函数
  function preorder(node,deep){
    // 如果当前节点为 null,退出递归
    if(node === null) return;
    // 如果当前深度对应子数组未创建,创建子数组
    if(!res[deep]) res[deep] = [];
    // 如果当前深度为奇数,从右向左记录节点值
    if(deep%2) res[deep].unshift(node.val);
    // 如果当前深度为偶数,从左向右记录节点值
    else res[deep].push(node.val);
    // 递归处理左右子树
    preorder(node.left,deep+1);
    preorder(node.right,deep+1);
  }
  // 调用前序遍历方法
  preorder(root,0);
  // 返回结果数组
  return res;
};
复制代码

至此我们就完成了 leetcode-103-二叉树的锯齿形层序遍历

如有任何问题或建议,欢迎留言讨论!

猜你喜欢

转载自juejin.im/post/7041564626470846472