226. 翻转二叉树

1. 题目

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

示例 1：

img

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

示例 2：

img

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

示例 3：

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

提示：

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

2. 思路

  • 已知题目要求翻转二叉树的所有左右节点
  • 当进入函数时，判断root是否为空，如果为空时，则直接返回root
  • 交换root节点的left和right节点
  • 随后调用自身函数进行左右节点的递归翻转，传递入参为root.left和root.right
  • 完成后直接返回root节点

3. 代码

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
/**
 * 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 root;
        }

        var newRight = root.left;
        var newLeft = root.right;

        root.left = newLeft;
        root.right = newRight;

        invertTree(root.left);
        invertTree(root.right);

        return root;
    }
}

4. 复杂度

  • 时间复杂度：O(n)
  • 空间复杂度：O(1)

image-20231009100620511