js算法:翻转二叉树

1,242 阅读2分钟

这是我参与11月更文挑战的第13天,活动详情查看:2021最后一次更文挑战

背景

学习前端三个月了,准备刷刷面试题,总结总结,一天几道面试题,向大厂进军。

上篇文章我们已经讲解了树以及二叉树的基本知识:详解二叉树,计算二叉树的深度

今天我们来了解下翻转二叉树,这个还有一个比较有意思的故事:

谷歌:我们90%的工程师使用您编写的软件(Homebrew),但是您却无法在面试的时候在白板上写出翻转二叉树,这太糟糕了!

连Homebrew的作者都不会,我们要是学会,是不是nb Plus?

问题:翻转二叉树

解析

我们用图来解析一下什么是翻转二叉树,翻转是左右翻转,而不是上下翻转。

image.png

思路:

我们从根节点开始,递归的对树进行遍历,并从叶子节点(最外层叶子节点)开始进行左右交换,然后是内部节点(内部叶子节点)进行左右交换,最后遍历到根节点(root)的时候,只需要交换左右两颗子树就可以完成对整个二叉树的翻转。

我们来用gif图演示下效果(图片来源于 力扣):

abc.gif

代码

    /**
     * Definition for a binary tree node.
     * function TreeNode(val, left, right) {
     *     this.val = (val===undefined ? 0 : val)
     *     this.left = (left===undefined ? null : left)
     *     this.right = (right===undefined ? null : right)
     * }
     */
    /**
     * @param {TreeNode} root
     * @return {TreeNode}
     */
    var invertTree = function(root) {
        if(root==null) return null;
        //递归翻转左边,其实就是从最外层叶子节点开始翻转。形成左边子树。
        var left=invertTree(root.left);
        //递归翻转右边,其实就是从最外层叶子节点开始翻转。形成右边子树。
        var right=invertTree(root.right);
        //左右子树 交换
        root.left=right;
        root.right=left;
        return root;
    };

验证结果:

image.png

image.png

是不是感觉也没有那么难,了解原理,还是挺容易实现的。

结语

一步一步慢慢来,踏踏实实把活干!