代码随想录第15天|226.翻转二叉树

  226.翻转二叉树

226. 翻转二叉树 - 力扣（LeetCode）

代码随想录 (programmercarl.com)

听说一位巨佬面Google被拒了，因为没写出翻转二叉树 | LeetCode：226.翻转二叉树_哔哩哔哩_bilibili

给你一棵二叉树的根节点 root ，翻转这棵二叉树，并返回其根节点。

示例 1：

​

输入：root = [4,2,7,1,3,6,9]
输出：[4,7,2,9,6,3,1]

示例 2：

​

输入：root = [2,1,3]
输出：[2,3,1]

示例 3：

输入：root = []
输出：[]

提示：

• 树中节点数目范围在 [0, 100] 内
• -100 <= Node.val <= 100

刚开始的思考还挺混乱的，经过卡哥提点和强调，意识到翻转二叉树不仅仅要交换数值，更要改变指针指向。

​

递归三部曲：

1、确定递归的参数和返回值：参数就是传入节点的指针，返回的是为root节点的指针，所以返回值为void

public TreeNode invertTree(TreeNode root) {
 
}

2、确定终止条件:节点为空时终止遍历

// 如果当前节点为空，直接返回空
        if (root == null) {
            return null;
        }

3、确定单层遍历的逻辑：前序遍历的话就是先交换左右孩子节点，然后反转左子树，然后反转右子树

// 递归调用，先反转左子树
        invertTree(root.left);
        // 递归调用，再反转右子树
        invertTree(root.right);
        // 调用swapChildren方法交换当前节点的左右子树
        swapChildren(root);
        // 返回反转后的二叉树的根节点
        return root;
    }

    // 定义一个私有方法swapChildren，用于交换当前节点的左右子树
    private void swapChildren(TreeNode root) {
        // 交换当前节点的左右子树
        TreeNode tmp = root.left;
        root.left = root.right;
        root.right = tmp;
    }

综合代码：

// 定义一个名为Solution的类
class Solution {
    // 定义一个公共方法invertTree，用于反转二叉树，采用DFS递归实现
    public TreeNode invertTree(TreeNode root) {
        // 如果当前节点为空，直接返回空
        if (root == null) {
            return null;
        }
        // 递归调用，先反转左子树
        invertTree(root.left);
        // 递归调用，再反转右子树
        invertTree(root.right);
        // 调用swapChildren方法交换当前节点的左右子树
        swapChildren(root);
        // 返回反转后的二叉树的根节点
        return root;
    }

    // 定义一个私有方法swapChildren，用于交换当前节点的左右子树
    private void swapChildren(TreeNode root) {
        // 交换当前节点的左右子树
        TreeNode tmp = root.left;
        root.left = root.right;
        root.right = tmp;
    }
}