Leetcode之二叉树翻转

博主在面试字节跳动实习生岗位时，遇见了二叉树翻转的问题。而刷过 leetcode 的同学们对此也不会陌生，因为它就是 Leetcode 上的原题——invert-binary-tree。因此，博主在此给出解题思路和解题方案，希望能对大家有所帮助。

简介

题目要求十分简单——给定一棵二叉树，将其左右子树进行翻转，如下图所示。

从上图中，我们不难看出原二叉树结点的父子关系不变，兄弟关系发生了交换。那么，我们又如何解决呢？

思路

以前一小节的输入输出为例，经过分析，我们不难发现，如果在根节点值为2的子树和根节点值为7的子树完成翻转之后，我们只需改变它们的兄弟关系即可。换句话说，我们只需要把根节点值为4的子树的左孩子设置为原先的右孩子，右孩子设置为原先的左孩子即可完成二叉树的翻转。因此，我们可以利用递归的方式自底向上完成翻转，源代码如下。

代码

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

class Solution {
    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;
    }
}