Rayman的绝顶之路——Leetcode每日一题打卡20

Leetcode 226. 翻转二叉树 题目：

翻转一棵二叉树。

示例：

输入：

输出：

备注:

这个问题是受到 Max Howell 的 原问题 启发的 ：

谷歌：

我们90％的工程师使用您编写的软件(Homebrew)，但是您却无法在面试时在白板上写出翻转二叉树这道题，这太糟糕了。

思路：

首先这是一道很经典的二叉树问题。显然，我们从根节点开始，递归地对树进行遍历，并从叶子结点先开始翻转。如果当前遍历到的节点 root 的左右两棵子树都已经翻转，那么我们只需要交换两棵子树的位置，即可完成以 root 为根节点的整棵子树的翻转。

Java AC代码：

public class Leetcode226 {
    public class TreeNode {
        int val;
        TreeNode left;
        TreeNode right;
        TreeNode(int x) {
            val = x;
        }
    }

    public TreeNode invertTree(TreeNode root) {
        if (root == null) {
            return null;
        }
        TreeNode left = invertTree(root.left);
        TreeNode right = invertTree(root.right);
        root.left = right;
        root.right = left;
        return root;
    }
}

但是此时内存消耗没有超过50%以上的用户，向我精益求精的人怎么会就此停止呢。

于是在原来代码的基础上做了一些处理。

Java AC代码优化 ：

public class Leetcode226 {
    public class TreeNode {
        int val;
        TreeNode left;
        TreeNode right;

        TreeNode(int x) {
            val = x;
        }
    }

    public TreeNode invertTree(TreeNode root) {
        makeInvertTree(root);
        return root;
    }

    public static void makeInvertTree(TreeNode root) {
        if (root == null) {
            return;
        }
        TreeNode temp;
        if (root.left != null && root.right != null) {
            temp = root.left;
            root.left = root.right;
            root.right = temp;
            makeInvertTree(root.left);
            makeInvertTree(root.right);
        } else if (root.left != null && root.right == null) {
            temp = root.left;
            root.left = null;
            root.right = temp;
            makeInvertTree(root.right);
        } else if (root.left == null && root.right != null) {
            temp = root.right;
            root.right = null;
            root.left = temp;
            makeInvertTree(root.left);
        }
    }
}

虽然没有达到双百，但是90多还算可以。

2020.9.16打卡