关于LeetCode中Invert Binary Tree一题的理解

最新推荐文章于 2020-09-19 13:03:05 发布

原创最新推荐文章于 2020-09-19 13:03:05 发布 · 446 阅读

0 ·

CC 4.0 BY-SA版权

文章标签：

#leetcode

java 专栏收录该内容

60 篇文章

订阅专栏

本文介绍了一种经典的二叉树翻转算法，通过递归方式实现节点左右子树的交换，达到整体翻转的效果。文章详细解释了递归过程中的基线条件，并给出了具体的代码实现。

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >

题目如下：

Invert a binary tree.

     4
   /   \
  2     7
 / \   / \
1   3 6   9

     4
   /   \
  7     2
 / \   / \
9   6 3   1

Trivia:
This problem was inspired by this original tweet by Max Howell:

Google: 90% of our engineers use the software you wrote (Homebrew), but you can’t invert a binary tree on a whiteboard so fuck off.

非常有意思的问题描述，灵感来自于谷歌的一名大神，他写出了Homebrew，却不能在白板上将一个二叉树倒置......

言归正传，又是一道和二叉树有关的题目，说到二叉树几乎都绕不开“递归”二字，我们总是要通过遍历二叉树来获取相关信息，那么递归基本上就是我们不二的选择。要使用递归，我们就要找到基线，这里的基线基本就要定位在叶子节点和节点为null的情况。节点为叶子节点时，它的倒置（invert）就是它本身；如果该节点为null，那么它的倒置（invert）也是null。了解了基线情况后，我们就要开始分析一般情况，一般情况就是一个节点的子节点中其中至少有一个不为空，这里的倒置（invert）其实就是交换这两个字节点的位置，非常容易实现。有了上述思路之后，代码实现起来就很轻松了，如下所示：

<span style="font-size:14px;">/**
 * Definition for a binary tree node.
 * public class TreeNode {
 *     int val;
 *     TreeNode left;
 *     TreeNode right;
 *     TreeNode(int x) { val = x; }
 * }
 */
public class Solution {
    public TreeNode invertTree(TreeNode root) {
        if(root==null){
            return null;
        }else if(root.right==null && root.left==null){
            return root;
        }else{
            TreeNode temp = invertTree(root.left);
            root.left = invertTree(root.right);
            root.right = temp;
            return root;
        }
    }
}</span>

其实后来写完代码的时候发现else if中的情况其实可以合并到else中去，看来考虑问题的深入程度还有待提高啊。