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

秋招算法备战第22天 | 654.最大二叉树、617.合并二叉树、700.二叉搜索树中的搜索、98.验证二叉搜索树

235. 二叉搜索树的最近公共祖先 - 力扣(LeetCode)

在一个二叉搜索树中,两个节点 p 和 q 的最近公共祖先可以通过以下的算法找到:

  1. 从根节点开始。
  2. 如果当前节点的值大于 p 和 q 的值,那么你需要转向左子树。因为在二叉搜索树中,左子树的所有节点的值都比当前节点的值小。
  3. 如果当前节点的值小于 p 和 q 的值,那么你需要转向右子树。因为在二叉搜索树中,右子树的所有节点的值都比当前节点的值大。
  4. 如果以上条件都不满足,那么当前节点就是 p 和 q 的最近公共祖先。

在 Python 中,这个算法可以像这样实现:

class TreeNode:def __init__(self, x):self.val = xself.left = Noneself.right = Nonedef lowestCommonAncestor(root, p, q):while root:if root.val > p.val and root.val > q.val:root = root.leftelif root.val < p.val and root.val < q.val:root = root.rightelse:return rootreturn None

在这个代码中,lowestCommonAncestor 函数将接受一个二叉搜索树的根节点 root,和两个需要找最近公共祖先的节点 pq。然后它将返回 p 和 q 的最近公共祖先。如果没有找到,那么它将返回 None。

注意这个算法假设 p 和 q 都在这棵二叉搜索树中。

701. 二叉搜索树中的插入操作 - 力扣(LeetCode)

因为是二叉树,直接按照规则遍历,然后在对应叶子结点插入

# 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 insertIntoBST(self, root: Optional[TreeNode], val: int) -> Optional[TreeNode]:if root is None:return TreeNode(val)pre = Nonedef traversal(cur, val, left_child=False):nonlocal preif cur is None:if left_child:pre.left = TreeNode(val)else:pre.right = TreeNode(val)returnpre = curif cur.val > val:traversal(cur.left, val, left_child=True)else:traversal(cur.right, val)traversal(root, val)return root

下面是GPT4对该代码的优化

你的代码已经很好地实现了插入操作。你在遍历树的时候,递归地找到插入节点的位置,然后创建新节点并插入到正确的位置。同时,你用一个外部变量
pre 来跟踪上一个节点,以便在正确的位置插入新节点。

然而,你的代码可以被优化,使其更简洁和易于理解。在优化后的代码中,你可以使用返回值,而不是外部变量来跟踪插入的位置。这样,你可以避免使用
nonlocal 关键字,并使函数的逻辑更加清晰。以下是优化后的代码:

# 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 insertIntoBST(self, root: Optional[TreeNode], val: int) -> Optional[TreeNode]:if root is None:return TreeNode(val)if root.val > val:root.left = self.insertIntoBST(root.left, val)else:root.right = self.insertIntoBST(root.right, val)return root ```在优化后的代码中,插入新节点的逻辑被分配给递归调用的返回值,这使得代码更加简洁并且逻辑更加清晰。这个函数将创建一个新的 `TreeNode`
如果 `root` 是 `None`,否则它将递归地遍历左子树或右子树,根据 `val`
的值决定向哪个方向遍历。然后,递归调用的返回值(即新节点或现有节点)被分配给 `root.left` 或
`root.right`,从而实现插入操作。最后,函数返回 `root`,它现在可能有一个新的子节点。

450. 删除二叉搜索树中的节点 - 力扣(LeetCode)

对于二叉搜索树的删除操作,主要分为以下几个步骤:

  1. 找到需要删除的节点;
  2. 如果节点是叶子节点,可以直接删除;
  3. 如果节点只有一个孩子,可以直接将其父节点的相应链接指向它的孩子;
  4. 如果节点有两个孩子,需要找到其右子树的最小节点(或左子树的最大节点),用这个节点代替要删除的节点,然后删除这个最小(或最大)节点。

以下是一个用Python实现的例子:

# 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 = rightclass Solution:def deleteNode(self, root: Optional[TreeNode], key: int) -> Optional[TreeNode]:if not root:return Noneif key < root.val:  # key在左子树root.left = self.deleteNode(root.left, key)elif key > root.val:  # key在右子树root.right = self.deleteNode(root.right, key)else:  # 找到key,进行删除操作if not root.left:  # 左子树为空,直接返回右子树return root.rightif not root.right:  # 右子树为空,直接返回左子树return root.leftelse:  # 左右子树均不为空min_node = self.findMin(root.right)  # 找到右子树的最小节点root.val = min_node.val  # 将最小节点的值复制到rootroot.right = self.deleteNode(root.right, min_node.val)  # 删除右子树中的最小节点return rootdef findMin(self, node: Optional[TreeNode]) -> Optional[TreeNode]:# 找到树中的最小节点while node.left:node = node.leftreturn node

这个函数的思路主要是使用递归来找到要删除的节点,然后按照上面提到的规则删除。在这个过程中,我们使用了一个辅助函数 findMin 来找到一个树中的最小节点。

总结

今天的题主要是将结点作为返回值,通过遍历过程中更新左右子树来完成对树的操作

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

相关文章:

  • 小程序之移花宫-自定义底部标签图标---【浅入深出系列005】
  • 题目1 SQL注入(保姆级教程)
  • PDF转换成word乱码了怎么办?最实用的方法在这里!
  • 字节跳动后端面试,笔试部分
  • [JavaScript游戏开发] 2D二维地图绘制、人物移动、障碍检测
  • 区间预测 | MATLAB实现基于QRF随机森林分位数回归时间序列区间预测模型
  • .NET网络编程——TCP通信
  • 【Python机器学习】实验01 Numpy以及可视化回顾
  • vue3-组件中的变化
  • 认识主被动无人机遥感数据、预处理无人机遥感数据、定量估算农林植被关键性状、期刊论文插图精细制作与Appdesigner应用开发
  • 数学建模的六个步骤
  • 【计算机组成原理】24王道考研笔记——第二章 数据的表示和运算
  • JQ-6 Bootstrap入门到实战;Bootstrap的(优缺点、安装、响应式容器原理、网格系统、响应式工具类、Bootstrap组件);小项目实践
  • 如何用3D格式转换工具HOOPS Exchange读取颜色和材料信息?
  • [Ubuntu 22.04] 安装docker,并设置镜像加速
  • 如何使用GPT作为SQL查询引擎的自然语言
  • Servlet3.0上传文件
  • 【ARM Cache 系列文章 6 番外篇 – MMU, MPU, SMMU, PMU 差异与关系】
  • NetSuite ERP顾问的进阶之路
  • js 新浏览器打开页面
  • jmeter软件测试实验(附源码以及配置)
  • ZooKeeper原理剖析
  • 【算组合数】CF1833 F
  • Attention详解(自用)
  • pptx转pdf工具类
  • 2023华为OD统一考试(B卷)题库清单(持续收录中)以及考点说明
  • 论文笔记--Won’t Get Fooled Again: Answering Questions with False Premises
  • 【Django】include app_name和namespace的区别
  • (黑客)自学笔记
  • 【期末课程设计】学生成绩管理系统