算法题解-翻转二叉树

162 阅读2分钟

题目

给你一棵二叉树的根节点 root ,翻转这棵二叉树,并返回其根节点。

输入: root = [4,2,7,1,3,6,9]
输出: [4,7,2,9,6,3,1]

题解

第一种

在处理二叉树问题时十分适合使用递归操作,所以我们这里采用递归,我们这里先判断根节点是否为null,如果是则我们直接返回null,否则我们去递归调用invertTree函数来翻转根节点的左右子树,我们把递归调用的结果分别存储在Left和Right变量中,整个过程中我们都会不断的进行调用递归函数直到遍历到叶子节点为止,在每一层递归中,我们都会去翻转当前节点的左右子树,然后我们将根节点的左子树指向Right,右子树指向Left,这样我们就完成了翻转操作,最后我们在返回翻转后的根节点即可

var invertTree = function(root) {
    if(root == null){
        return null
    }
    else{
        const Left = invertTree(root.left)
        const Right = invertTree(root.right)
        root.left = Right
        root.right = Left
        return root
    }
};

第二种

我们在函数中先判断root是否为空是则直接返回root,然后我们在声明一个变量queue,并将根节点root放入其中,然后我们使用循环去遍历queue,循环条件为queue为空,在循环中,我们先获取当前queue变量的长度,并且用变量len保存起来,然后我们在使用一个循环来处理当前层级的节点,在循环中,我们先从queue变量中取出一个节点在用变量node保存起来,然后我们通过交换节点的左右子节点,实现节点的翻转,这里使用了Es6中解构赋值的方式,我们将node.left和node.right互换位置,然后我们在判断当前节点的左子节点如果不为空,我们则将左子节点加入队列中,再判断当前节点的右子节点如果不为空,我们则将右子节点加入队列中,当完成当前节点的处理后,我们就继续进行下一个节点的处理,直到当前层级的所有节点都被处理完,当循环结束后,我们将root返回出去即可

var invertTree = function(root) {
  if(root === null) return root
  let queue = [root]
  while(queue.length) {
    let len = queue.length
    for(let i = 0; i < len; i++) {
      let node = queue.shift();
      [node.left, node.right] = [node.right, node.left];
      node.left && queue.push(node.left);
      node.right && queue.push(node.right);
    }
  }
  return root
};

坚持努力,无惧未来!