0x00 题目
给定一个【二叉搜索树】, 找到该树中两个指定节点的最近公共祖先
对于有根树 T 的两个结点 p、q,最近公共祖先表示为一个结点 x,满足 x 是 p、q 的祖先且 x 的深度尽可能大(一个节点也可以是它自己的祖先)
0x01 思路
一说到 二叉搜索树,就应该马上想到 中序遍历,结果是 升序 的
另外一个特征是:根节点 大于 左侧 所有节点,小于 右侧 所有节点
0x02 解法
语言:Swift
树节点:TreeNode
public class TreeNode {
public var val: Int
public var left: TreeNode?
public var right: TreeNode?
public init() { self.val = 0; self.left = nil; self.right = nil; }
public init(_ val: Int) { self.val = val; self.left = nil; self.right = nil; }
public init(_ val: Int, _ left: TreeNode?, _ right: TreeNode?) {
self.val = val
self.left = left
self.right = right
}
}
递归方式:
func lowestCommonAncestor(_ root: TreeNode?, _ p: TreeNode?, _ q: TreeNode?) -> TreeNode? {
guard let root = root else {
return nil
}
// 比根节点都大, 往右找
if root.val < p!.val && root.val < q!.val {
return lowestCommonAncestor(root.right, p, q)
}
// 比根节点都小,往左找
else if root.val > p!.val && root.val > q!.val {
return lowestCommonAncestor(root.left, p, q)
}else{
return root
}
}
0x03 我的小作品
欢迎体验我的作品之一:小编辑器-XCompiler
小巧的在线编辑器!
App Store 搜索即可~