【算法题解】31. 翻转二叉树的递归解法

news2025/4/17 4:54:36

这是一道简单题
https://leetcode.cn/problems/invert-binary-tree/

题目

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

示例 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 <= N o d e . v a l <= 100$

题解

翻转二叉树，就是交换根结点的左右子节点，而左右子节点同样需要进行翻转，也就是说需要交换翻转后的左右子节点。

以Java代码为例：

TreeNode left = root.left;
TreeNode right = root.right;
// 交换翻转后的左右子节点
root.left = invertTree(right);
root.right = invertTree(left);

以上代码片段明显可以看出使用的是递归的思路。既然使用递归，我们就需要确定递归的三要素：

递归函数：交换根结点的翻转后的左右子节点。
边界条件：根节点为空，无需交换。
还原现场：无需还原。

Java 代码实现

/**
 * 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){
            return null;
        }
        TreeNode left = root.left;
        TreeNode right = root.right;
        root.left = invertTree(right);
        root.right = invertTree(left);
        return root;
    }
}

Go 代码实现

/**
 * Definition for a binary tree node.
 * type TreeNode struct {
 *     Val int
 *     Left *TreeNode
 *     Right *TreeNode
 * }
 */
func invertTree(root *TreeNode) *TreeNode {
    
    if root == nil {
        return nil
    }

    left, right := root.Left, root.Right

    root.Left = invertTree(right)
    root.Right = invertTree(left)

    return root
    
}