[算法]二叉搜索树的最近公共祖先

89 阅读1分钟

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 搜索即可~