zl程序教程

您现在的位置是:首页 >  其他

当前栏目

LeetCode 226. 翻转二叉树

2023-09-14 09:13:12 时间

LeetCode 226. 翻转二叉树

难度: e a s y \color{Green}{easy} easy


题目描述

给你一棵二叉树的根节点 r o o t root 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 ] [0, 100] [0,100]
  • − 100 < = N o d e . v a l < = 100 -100 <= Node.val <= 100 100<=Node.val<=100

算法

(递归)

我们从根节点开始,递归地对树进行遍历,并从叶子节点先开始翻转。如果当前遍历到的节点 root 的左右两棵子树都已经翻转,那么我们只需要交换两棵子树的位置,即可完成以 root 为根节点的整棵子树的翻转。

复杂度分析

  • 时间复杂度 O ( n ) O(n) O(n),其中 n n n 为二叉树节点的数目。我们会遍历二叉树中的每一个节点,对每个节点而言,我们在常数时间内交换其两棵子树

  • 空间复杂度 : O ( n ) O(n) O(n)

C++ 代码

/**
 * Definition for a binary tree node.
 * struct TreeNode {
 *     int val;
 *     TreeNode *left;
 *     TreeNode *right;
 *     TreeNode() : val(0), left(nullptr), right(nullptr) {}
 *     TreeNode(int x) : val(x), left(nullptr), right(nullptr) {}
 *     TreeNode(int x, TreeNode *left, TreeNode *right) : val(x), left(left), right(right) {}
 * };
 */
class Solution {
public:
    TreeNode* invertTree(TreeNode* root) {
        if (!root) return NULL;
        swap(root->left, root->right);
        invertTree(root->left);
        invertTree(root->right);
        return root;
    }
};