无需手写代码，助你快速找到二叉树的最近公共祖先

闲谈

2023-10-29 13:27:59

相信很多程序员都遇到过这样一个问题：需要在二叉树中找到两个节点的最近公共祖先。手写代码来解决这个问题既耗时又容易出错。别担心，今天我来介绍一个更简单的方法，让你无需一行代码就能轻松搞定！

借助 AI 螺旋，轻松实现高效寻祖

AI 螺旋是一款功能强大的 AI 工具，它可以帮助我们自动化许多繁琐的任务。其中一项重要的功能就是可以快速找到二叉树的最近公共祖先。

使用 AI 螺旋查找最近公共祖先非常简单，只需要几个简单的步骤：

首先，在 AI 螺旋中创建一个新项目。
选择“算法和数据结构”类别。
在搜索框中输入“二叉树最近公共祖先”。
点击“创建”。

AI 螺旋将自动生成一个解决方案，其中包含用于查找最近公共祖先的代码。代码是 Python 编写的，非常容易理解和使用。

具体代码详解

def find_lca(root, p, q):
    if root is None:
        return None

    if root.val == p.val or root.val == q.val:
        return root

    left_lca = find_lca(root.left, p, q)
    right_lca = find_lca(root.right, p, q)

    if left_lca and right_lca:
        return root

    return left_lca if left_lca else right_lca

使用示例

from binary_tree import TreeNode

root = TreeNode(1)
root.left = TreeNode(2)
root.right = TreeNode(3)
root.left.left = TreeNode(4)
root.left.right = TreeNode(5)
root.right.left = TreeNode(6)
root.right.right = TreeNode(7)

p = root.left.left
q = root.right.right

lca = find_lca(root, p, q)
print(lca.val)  # 输出：1