翻转二叉树

力扣 226. 翻转二叉树

题目说明

翻转一棵二叉树。

示例：

输入：

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

输出：

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

笔者理解

此题是一道二叉树算法问题，在力扣题库中被定义为简单题。

解法

当笔者阅读完此题后，采用了递归翻转左右子树的方式，让我们来看看具体如何实现的吧。

实现

/**
 * Definition for a binary tree node.
 * public class TreeNode {
 *     int val;
 *     TreeNode left;
 *     TreeNode right;
 *     TreeNode() {}
 *     TreeNode(int val) { this.val = val; }
 *     TreeNode(int val, TreeNode left, TreeNode right) {
 *         this.val = val;
 *         this.left = left;
 *         this.right = right;
 *     }
 * }
 */
class Solution {
    /*
     * 二叉树
     */
    public TreeNode invertTree(TreeNode root) {
        // 节点为空或者没有左右子节点时不需要翻转
        if (root == null || (root.left == null && root.right == null)) {
            return root;
        }
        // 交换左右子树
        TreeNode temp = root.left;
        root.left = root.right;
        root.right = temp;

        // 再翻转左右子树
        root.left = invertTree(root.left);
        root.right = invertTree(root.right);

        return root;
    }
}