翻转二叉树

79 阅读1分钟

开启掘金成长之旅!这是我参与「掘金日新计划 · 2 月更文挑战」的第 19 天,点击查看活动详情

一、题目描述

这是 LeetCode 上的第二百二十六题:翻转二叉树,难度为 简单

Tag:「二叉树」、「深度优先搜索」、「广度优先搜索」

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

示例 1:

image.png

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

示例 2:

image.png

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

示例 3:

输入: root = []
输出: []

提示:

1、树中节点数目范围在 [0, 100] 内

2、-100 <= Node.val <= 100

二、解题思路

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

代码实现:

class Solution {
    public TreeNode invertTree(TreeNode root) {
        if (root == null) {
            return null;
        }
        TreeNode left = invertTree(root.left);
        TreeNode right = invertTree(root.right);
        root.left = right;
        root.right = left;
        return root;
    }
}

复杂度分析

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

2、空间复杂度:O(N)。使用的空间由递归栈的深度决定,它等于当前节点在二叉树中的高度。在平均情况下,二叉树的高度与节点个数为对数关系,即 O(logN)。而在最坏情况下,树形成链状,空间复杂度为 O(N)。

三、总结

本道算法题难度为简单,使用递归我们能够快速的解决问题。另外递归的核心在于:

1、一个问题可以分解成具有相同解决思路的子问题,子子问题,换句话说这些问题都能调用同一个函数

2、递归需要有一个终止条件,如果没有的话,就无穷无尽地分解子问题了,问题显然是无解的。

所以解递归题的关键在于我们首先需要根据以上递归的两个特点判断题目是否可以用递归来解。

好了,本篇文章到这里就结束了,感谢你的阅读🤝