当前位置： 首页 > news >正文

算法训练营day18 530.二叉搜索树的最小绝对差、501.二叉搜索树中的众数、236. 二叉树的最近公共祖先

news 2025/7/15 6:05:48

        二叉树day6！本篇博客有很多二叉搜索树的题目，在这篇博客中，你会感觉到根据二叉树的性质、题目背景不同，适用的遍历方式也存在不同，遍历方式很重要

530.二叉搜索树的最小绝对差

        二叉搜索树采用中序遍历是一个有序数组。最直观的想法是：在一个有序数组（二叉搜索树转换）上求两个数最小差值，然后遍历数组，统计最小差值。遇到在二叉搜索树上求什么最值，求差值之类的，都要思考一下二叉搜索树可是有序的，要利用好这一特点。

递归（利用中序递增，结合数组）

# Definition for a binary tree node.
# class TreeNode:
#     def __init__(self, val=0, left=None, right=None):
#         self.val = val
#         self.left = left
#         self.right = right
class Solution:def getMinimumDifference(self, root: Optional[TreeNode]) -> int:self.vec = []self.traversal(root)if len(self.vec) < 2:return 0result = float('inf')for i in range(1,len(self.vec)):result = min(result, self.vec[i] - self.vec[i - 1])return resultdef traversal(self, root):if root is None:return self.traversal(root.left)self.vec.append(root.val)self.traversal(root.right)

递归法（记录前节点指针,同时计算）

        要学会在递归遍历的过程中记录前后两个指针，这也是一个小技巧

# Definition for a binary tree node.
# class TreeNode:
#     def __init__(self, val=0, left=None, right=None):
#         self.val = val
#         self.left = left
#         self.right = right
class Solution:def getMinimumDifference(self, root: Optional[TreeNode]) -> int:self.result = float('inf')self.pre = Noneself.traversal(root)return self.resultdef traversal(self, cur):if cur is None:returnself.traversal(cur.left)# 严格中序遍历顺序if self.pre is not None:self.result = min(self.result, cur.val - self.pre.val)self.pre = curself.traversal(cur.right)

迭代法

# Definition for a binary tree node.
# class TreeNode:
#     def __init__(self, val=0, left=None, right=None):
#         self.val = val
#         self.left = left
#         self.right = right
class Solution:def getMinimumDifference(self, root: Optional[TreeNode]) -> int:stack = []cur = rootpre = Noneresult = float('inf')while cur is not None or len(stack) > 0:if cur is not None:stack.append(cur)cur = cur.left # 左else:cur = stack.pop()if pre is not None: # 中result = min(result, cur.val - pre.val)pre = curcur = cur.right # 右return  result

501.二叉搜索树中的众数

        如果不是二叉搜索树：最直观的方法是把这个树进行遍历，用map统计频率，把频率排个序，最后取前面高频的元素的集合。

        如果是二叉搜索树：它中序遍历是有序的。频率count 等于 maxCount（最大频率），将元素加入到结果集中。频率count 大于 maxCount的时候，不仅要更新maxCount，而且要清空结果集（以下代码为result数组），因为结果集之前的元素都失效了。要不然就要遍历两次二叉搜索树才能把这个最高出现频率元素的集合求出来。

递归法（字典）

# Definition for a binary tree node.
# class TreeNode:
#     def __init__(self, val=0, left=None, right=None):
#         self.val = val
#         self.left = left
#         self.right = right
from collections import defaultdict
class Solution:def findMode(self, root: Optional[TreeNode]) -> List[int]:freq_map = defaultdict(int)# 创建一个字典, 自动处理不存在的键#并将其默认值设为 0, 从而简化计数和频率统计的代码。result = []if root is None:return resultself.searchBST(root, freq_map)max_freq = max(freq_map.values()) # 获取字典中所有值的最大值# 若 freq_map = {'a': 3, 'b': 5, 'c': 2}# 则 freq_map.values() 的结果为 dict_values([3, 5, 2])。for key, freq in freq_map.items():# 同时获取字典的键（key）和对应的值（value）if freq == max_freq:result.append(key)return resultdef searchBST(self, cur, freq_map):    if cur is None:returnfreq_map[cur.val] += 1self.searchBST(cur.left, freq_map)self.searchBST(cur.right, freq_map)# 前序遍历

递归法（利用二叉搜索树性质）

        指针的巧妙使用，同时也是值的巧妙覆盖

# Definition for a binary tree node.
# class TreeNode:
#     def __init__(self, val=0, left=None, right=None):
#         self.val = val
#         self.left = left
#         self.right = right
from collections import defaultdict
class Solution:def __init__(self):self.maxCount = 0self.count = 0self.pre = Noneself.result = []def findMode(self, root: Optional[TreeNode]) -> List[int]:self.count = 0self.maxCount = 0self.pre = Noneself.result = []self.searchBST(root)return self.resultdef searchBST(self, cur):if cur is None:return self.searchBST(cur.left) # 左# 中if self.pre is None: # 第一个节点self.count += 1elif self.pre.val == cur.val:self.count += 1else: # 与前节点不同self.count = 1self.pre = cur # 前节点指针更新if self.count == self.maxCount:self.result.append(cur.val)if self.count > self.maxCount:self.maxCount = self.countself.result = [cur.val] # 同时清空result, 清空这一步很关键！self.searchBST(cur.right) # 右return # 注意中序遍历顺序, 很关键

迭代法

        其实和递归区别不是很大，但是很多细节的设计写出来是不容易的

# Definition for a binary tree node.
# class TreeNode:
#     def __init__(self, val=0, left=None, right=None):
#         self.val = val
#         self.left = left
#         self.right = right
from collections import defaultdict
class Solution:def findMode(self, root: Optional[TreeNode]) -> List[int]:st = []cur = rootpre = NonemaxCount = 0 # 最大频率count = 0result = []while cur is not None or st: if cur is not None: # 指针来访问节点，访问到最底层st.append(cur)cur = cur.left # 左else:cur = st.pop() # 中if pre is None:count = 1elif pre.val == cur.val:count += 1else:count = 1 # 数值不同if count == maxCount:result.append(cur.val)if count > maxCount:maxCount = countresult = [cur.val]pre = curcur = cur.right # 右return result

236. 二叉树的最近公共祖先

        遇到这个题目首先想的是——能自底向上查找，这样就可以找到公共祖先了。那么二叉树如何可以自底向上查找呢？答案是——回溯，二叉树回溯的过程就是从底到上。后序遍历（左右中）就是天然的回溯过程，可以根据左右子树的返回值，来处理中节点的逻辑。

        首先最容易想到的一个情况（重点理解）：如果找到一个节点，发现左子树出现结点p，右子树出现节点q，或者 左子树出现结点q，右子树出现节点p，那么该节点就是节点p和q的最近公共祖先。

递归三部曲：

  • 确定递归函数返回值以及参数

        需要递归函数返回值，来告诉我们是否找到节点q或者p，那么返回值为bool类型就可以了。但我们还要返回最近公共节点，可以利用上题目中返回值是TreeNode * ，那么如果遇到p或者q，就把q或者p返回，返回值不为空，就说明找到了q或者p。

  • 确定终止条件

        遇到空的话，因为树都是空了，所以返回空。如果 root == q，或者 root == p，说明找到 q、p ，则将其返回，这个返回值，后面在中节点的处理过程中会用到

  • 确定单层递归逻辑

        值得注意的是 本题函数有返回值，是因为回溯的过程需要递归函数的返回值做判断（重点理解），但本题我们依然要遍历树的所有节点。

# Definition for a binary tree node.
# class TreeNode:
#     def __init__(self, x):
#         self.val = x
#         self.left = None
#         self.right = Noneclass Solution:def lowestCommonAncestor(self, root: 'TreeNode', p: 'TreeNode', q: 'TreeNode') -> 'TreeNode':if root == q or root == p or root is None: # 找到q or p, 或者没有找到(返回none)return rootleft = self.lowestCommonAncestor(root.left, p , q)right = self.lowestCommonAncestor(root.right, p , q)# 递归左子树：查找 p 或 q 是否存在于左子树中。# 递归右子树：查找 p 或 q 是否存在于右子树中。if left is not None and right is not None: # 左右子树均不为空：说明 p 和 q 分别位于当前节点的两侧return root # 这个判断需要好好理解！！if left is None and right is not None: # 仅左子树不为空：说明 p 和 q 都在左子树中, 相反在右子树中return rightelif left is not None and right is None: return leftelse:# 左右子树均为空：说明当前路径不包含 p 和 q，return None

那么我给大家归纳如下三点：

  1. 求最小公共祖先，需要从底向上遍历，那么二叉树，只能通过后序遍历（即：回溯）实现从底向上的遍历方式。

  2. 在回溯的过程中，必然要遍历整棵二叉树，即使已经找到结果了，依然要把其他节点遍历完，因为要使用递归函数的返回值（也就是代码中的left和right）做逻辑判断。

  3. 要理解如果返回值left为空，right不为空为什么要返回right，为什么可以用返回right传给上一层结果——两个指定节点，即节点必定存在

注释：引用自《代码随想录》

http://www.lryc.cn/news/587586.html

相关文章：

  • B站自动回复工具（破解）
  • 项目一第一天
  • 苍穹外卖学习指南（java的一个项目）（老师能运行，但你不行，看这里！！）
  • priority_queue的使用和模拟实现以及仿函数
  • 《C++内存泄漏8大战场：Qt/MFC实战详解 + 面试高频陷阱破解》
  • MFC/C++语言怎么比较CString类型最后一个字符
  • 【Linux】Ubuntu22.04安装zabbix
  • HTTP 四种常见方法
  • 【论文阅读】Thinkless: LLM Learns When to Think
  • ubuntu(22.04)系统上安装 MuJoCo
  • MFC UI控件CheckBox从专家到小白
  • 桶排序算法深度剖析
  • FastAPI + gRPC 全栈实践：Windows 开发到 Ubuntu 部署全指南
  • flink 和 spark 架构的对比
  • idea删除的文件怎么找回
  • IDEA中使用Servlet，tomcat输出中文乱码
  • JMeter 连接与配置 ClickHouse 数据库
  • 递推预处理floor(log_2{n})
  • 【脚本系列】如何使用 Python 脚本对同一文件夹中表头相同的 Excel 文件进行合并
  • uniapp video视频全屏播放后退出，页面字体变大，样式混乱问题
  • 基于Spring Boot的生活用品电商网站的设计与实现
  • 国内隧道IP代理技术解析：原理、优势与实战应用
  • 算法学习笔记：21.动态规划——从原理到实战，涵盖 LeetCode 与考研 408 例题
  • linux 文件搜索与文件内容查看
  • Imx6ull用网线与电脑连接
  • 游戏玩法的专利博弈
  • 11、鸿蒙Harmony Next开发：列表布局 (List)
  • Spark 和 Hadoop MapReduce 的基本概念及区别
  • Spring Boot项目结构解析：构建高效、清晰的代码框架
  • UE5多人MOBA+GAS 22、创建技能图标UI，实现显示蓝耗，冷却，以及数字显示的倒数计时还有雷达显示的倒数计时