力扣156题最全解法:如何上下翻转二叉树(递归与迭代方法详解,附图解)

简介: 力扣156题最全解法:如何上下翻转二叉树(递归与迭代方法详解,附图解)

❤️❤️❤️ 欢迎来到我的博客。希望您能在这里找到既有价值又有趣的内容,和我一起探索、学习和成长。欢迎评论区畅所欲言、享受知识的乐趣!

期待与您一起探索技术、持续学习、一步步打怪升级 欢迎订阅本专栏❤️❤️

在本篇文章中,我们将详细解读力扣第156题“上下翻转二叉树”。通过学习本篇文章,读者将掌握如何使用递归和迭代的方法来解决这一问题。每种方法都将配以详细的解释和图解,以便于理解。

问题描述

力扣第156题“上下翻转二叉树”描述如下:

给定一个二叉树,将其上下翻转使得原来的右子节点变成新的根节点,并且所有的左子节点依次变成新的右子节点,直到原来的左子节点成为新的叶节点。要求返回新的根节点。

示例:

给定的二叉树:
    1
   / \
  2   3
 / \
4   5
翻转后的二叉树:
    4
   / \
  5   2
       / \
      3   1

解题思路

  1. 初步分析
  • 翻转后的树结构较复杂,需要重新定义节点的左右子树。
  • 可以使用递归和迭代的方法实现树的翻转。
  1. 多种解法
  • 递归法:利用递归函数来实现树的翻转。
  • 迭代法:利用栈或队列来实现树的翻转。

递归法

解法思路

递归法通过递归调用函数来处理每个节点,并将当前节点的左子节点变成新的右子节点,右子节点变成新的根节点。

步骤
  1. 如果当前节点为空或当前节点的左子节点为空,直接返回当前节点。
  2. 递归处理当前节点的左子节点,得到新的根节点。
  3. 将当前节点的左子节点的左子节点指向当前节点的右子节点。
  4. 将当前节点的左子节点的右子节点指向当前节点。
  5. 将当前节点的左子节点和右子节点置为空。
  6. 返回新的根节点。
代码实现
class TreeNode:
    def __init__(self, val=0, left=None, right=None):
        self.val = val
        self.left = left
        self.right = right
def upsideDownBinaryTree(root: TreeNode) -> TreeNode:
    if not root or not root.left:
        return root
    new_root = upsideDownBinaryTree(root.left)
    root.left.left = root.right
    root.left.right = root
    root.left = None
    root.right = None
    return new_root
# 测试案例
def printTree(root: TreeNode):
    if root:
        print(root.val, end=' ')
        printTree(root.left)
        printTree(root.right)
# 构建示例二叉树
root = TreeNode(1)
root.left = TreeNode(2)
root.right = TreeNode(3)
root.left.left = TreeNode(4)
root.left.right = TreeNode(5)
# 翻转二叉树
new_root = upsideDownBinaryTree(root)
printTree(new_root)  # 输出: 4 5 2 3 1
图解
初始二叉树:
    1
   / \
  2   3
 / \
4   5
递归翻转步骤:
1. 处理节点 1:
   1
  / \
 2   3
/ \
4   5
2. 处理节点 2:
    2
   / \
  4   5
3. 处理节点 4:
4 为空,递归返回
4. 节点 2 的新根为 4:
    4
   / \
  5   2
       / \
      3   1

迭代法

解法思路

迭代法通过栈或队列来模拟递归过程,逐步处理每个节点,实现树的翻转。

步骤
  1. 初始化当前节点 curr 为根节点 rootprevNonenextNonetempNone
  2. 迭代处理每个节点:
  • next 指向当前节点的左子节点。
  • 将当前节点的左子节点指向 temp
  • temp 指向当前节点的右子节点。
  • 将当前节点的右子节点指向 prev
  • 更新 prev 为当前节点,currnext
  1. 返回 prev 作为新的根节点。
代码实现
def upsideDownBinaryTreeIterative(root: TreeNode) -> TreeNode:
    curr = root
    prev = None
    next = None
    temp = None
    while curr:
        next = curr.left
        curr.left = temp
        temp = curr.right
        curr.right = prev
        prev = curr
        curr = next
    return prev
# 测试案例
root = TreeNode(1)
root.left = TreeNode(2)
root.right = TreeNode(3)
root.left.left = TreeNode(4)
root.left.right = TreeNode(5)
new_root = upsideDownBinaryTreeIterative(root)
printTree(new_root)  # 输出: 4 5 2 3 1
图解
初始二叉树:
    1
   / \
  2   3
 / \
4   5
迭代翻转步骤:
1. curr = 1, prev = None
2. next = 2, curr.left = None, temp = 3, curr.right = None
3. prev = 1, curr = 2
    1
     \
      3
4. next = 4, curr.left = 3, temp = 5, curr.right = 1
5. prev = 2, curr = 4
    2
   / \
  5   1
       \
        3
6. next = None, curr.left = 5, temp = None, curr.right = 2
7. prev = 4, curr = None
    4
   / \
  5   2
       / \
      3   1

复杂度分析

  • 递归法
  • 时间复杂度:O(N),其中 N 是树中节点的数量。每个节点只访问一次。
  • 空间复杂度:O(N),最坏情况下(树呈链状),递归栈的深度为 N。
  • 迭代法
  • 时间复杂度:O(N),其中 N 是树中节点的数量。每个节点只访问一次。
  • 空间复杂度:O(1),只使用了常数空间来存储临时变量。

测试案例分析

  1. 测试案例 1
  • 输入:
1
   / \
  2   3
 / \
4   5
  • 输出:
4
   / \
  5   2
       / \
      3   1
  • 解释:通过递归或迭代方法可以验证树的正确翻转。
  1. 测试案例 2
  • 输入:
1
     / 
    2
   /
  3
  • 输出:
3
 / \
2   1
  • 解释:处理单链的树,确保翻转正确。

总结

本文详细解读了力扣第156题“上下翻转二叉树”,通过递归法和迭代法两种不同的解法,帮助读者深入理解如何高效地实现树的翻转。希望读者通过本文的学习,能够在力扣刷题的过程中更加得心应手。

参考资料

  • 《算法导论》—— Thomas H. Cormen, Charles E. Leiserson, Ronald L. Rivest, Clifford Stein
  • 力扣官方题解

🌹🌹如果觉得这篇文对你有帮助的话,记得一键三连关注、赞👍🏻、收藏是对作者最大的鼓励,非常感谢 ❥(^_-)

❤️❤️关注公众号 数据分析螺丝钉 回复 学习资料 领取高价值免费学习资料❥(^_-)

欢迎关注微信公众号 数据分析螺丝钉

相关文章
|
2天前
|
Go 开发者 索引
【LeetCode 热题100】路径与祖先:二叉树中的深度追踪技巧(力扣33 / 81/ 153/154)(Go语言版)
本文深入探讨了LeetCode中四道关于「搜索旋转排序数组」的经典题目,涵盖了无重复和有重复元素的情况。通过二分查找的变形应用,文章详细解析了每道题的解题思路和Go语言实现代码。关键点包括判断有序区间、处理重复元素以及如何缩小搜索范围。文章还总结了各题的异同,并推荐了类似题目,帮助读者全面掌握二分查找在旋转数组中的应用。无论是初学者还是有经验的开发者,都能从中获得实用的解题技巧和代码实现方法。
30 14
|
1月前
|
Go
【LeetCode 热题100】路径与祖先:二叉树中的深度追踪技巧(力扣437 / 236 )(Go语言版)
本文深入探讨二叉树中路径与祖先问题,涵盖两道经典题目:LeetCode 437(路径总和 III)和236(最近公共祖先)。对于路径总和 III,文章分析了双递归暴力解法与前缀和优化方法,后者通过哈希表记录路径和,将时间复杂度从O(n²)降至O(n)。在最近公共祖先问题中,采用后序遍历递归查找,利用“自底向上”的思路确定最近公共祖先节点。文中详细解析代码实现与核心要点,帮助读者掌握深度追踪技巧,理解树结构中路径与节点关系的本质。这类问题在面试中高频出现,掌握其解法意义重大。
52 4
|
1月前
|
算法 Go
【LeetCode 热题100】深入理解二叉树结构变化与路径特性(力扣104 / 226 / 114 / 543)(Go语言版)
本博客深入探讨二叉树的深度计算、结构变换与路径分析,涵盖四道经典题目:104(最大深度)、226(翻转二叉树)、114(展开为链表)和543(二叉树直径)。通过递归与遍历策略(前序、后序等),解析每题的核心思路与实现方法。结合代码示例(Go语言),帮助读者掌握二叉树相关算法的精髓。下一讲将聚焦二叉树构造问题,欢迎持续关注!
58 10
|
1月前
|
存储 算法 数据可视化
【二叉树遍历入门:从中序遍历到层序与右视图】【LeetCode 热题100】94:二叉树的中序遍历、102:二叉树的层序遍历、199:二叉树的右视图(详细解析)(Go语言版)
本文详细解析了二叉树的三种经典遍历方式:中序遍历(94题)、层序遍历(102题)和右视图(199题)。通过递归与迭代实现中序遍历,深入理解深度优先搜索(DFS);借助队列完成层序遍历和右视图,掌握广度优先搜索(BFS)。文章对比DFS与BFS的思维方式,总结不同遍历的应用场景,为后续构造树结构奠定基础。
136 10
|
1月前
|
Go 索引 Perl
【LeetCode 热题100】【二叉树构造题精讲:前序 + 中序建树 & 有序数组构造 BST】(详细解析)(Go语言版)
本文详细解析了二叉树构造的两类经典问题:通过前序与中序遍历重建二叉树(LeetCode 105),以及将有序数组转化为平衡二叉搜索树(BST,LeetCode 108)。文章从核心思路、递归解法到实现细节逐一拆解,强调通过索引控制子树范围以优化性能,并对比两题的不同构造逻辑。最后总结通用构造套路,提供进阶思考方向,帮助彻底掌握二叉树构造类题目。
115 9
|
7月前
【LeetCode 43】236.二叉树的最近公共祖先
【LeetCode 43】236.二叉树的最近公共祖先
60 0
|
7月前
【LeetCode 38】617.合并二叉树
【LeetCode 38】617.合并二叉树
47 0
|
8月前
|
Unix Shell Linux
LeetCode刷题 Shell编程四则 | 194. 转置文件 192. 统计词频 193. 有效电话号码 195. 第十行
本文提供了几个Linux shell脚本编程问题的解决方案,包括转置文件内容、统计词频、验证有效电话号码和提取文件的第十行,每个问题都给出了至少一种实现方法。
117 6
LeetCode刷题 Shell编程四则 | 194. 转置文件 192. 统计词频 193. 有效电话号码 195. 第十行
|
9月前
|
Python
【Leetcode刷题Python】剑指 Offer 32 - III. 从上到下打印二叉树 III
本文介绍了两种Python实现方法,用于按照之字形顺序打印二叉树的层次遍历结果,实现了在奇数层正序、偶数层反序打印节点的功能。
98 6
|
9月前
|
搜索推荐 索引 Python
【Leetcode刷题Python】牛客. 数组中未出现的最小正整数
本文介绍了牛客网题目"数组中未出现的最小正整数"的解法,提供了一种满足O(n)时间复杂度和O(1)空间复杂度要求的原地排序算法,并给出了Python实现代码。
202 2

热门文章

最新文章