vlambda博客
学习文章列表

算法-链表010-二叉树及BST寻找最近公共祖先

                    二叉树的最近公共祖先

示例 1:

输入:root = [3,5,1,6,2,0,8,null,null,7,4], p = 5, q = 1
输出:3
解释:节点 5 和节点 1 的最近公共祖先是节点 3 。

示例 2:

算法-链表010-二叉树及BST寻找最近公共祖先


输入:root = [3,5,1,6,2,0,8,null,null,7,4], p = 5, q = 4
输出:5
解释:节点 5 和节点 4 的最近公共祖先是节点 5 。因为根据定义最近公共祖先节点可以为节点本身。
示例 3
输入:root = [1,2], p = 1, q = 2
输出:1

分析

此题找的是最近,如果左有,右边也有,最近的肯定是第一次就遇到的值是最近的

分几种情况

  1. 如果p、q都在左侧,那么遇到的提前遇到的肯定是最早的公共父节点
  2. 如果p、q都在右侧,那么最早遇到的节点一定是最早的公共父节点
  3. 如果p、q存在于左右两侧,那么最早的公共父节点一定是root节点

基于此规则,那么我们分别在左侧和右侧寻找

  • 如果左侧为空,那么说明都在右侧,返回右侧的第一个遇到的
  • 如果右侧为空,说明都在左侧,返回遇到的第一个
  • 如果左右都不为空,那么最早的一定是跟节点

==构造树结构==

type TreeNode struct {
 Val int
 Left *TreeNode
 Right *TreeNode
}

root1 := &TreeNode{
  Val:   3,
  Left:  &TreeNode{
   Val:   5,
   Left:  &TreeNode{
    Val:   6,
    Left:  nil,
    Right: nil,
   },
   Right: &TreeNode{
    Val:   2,
    Left:  &TreeNode{
     Val:   7,
     Left:  nil,
     Right: nil,
    },
    Right: &TreeNode{
     Val:   4,
     Left:  nil,
     Right: nil,
    },
   },
  },
  Right: &TreeNode{
   Val:   1,
   Left:  &TreeNode{
    Val:   0,
    Left:  nil,
    Right: nil,
   },
   Right: &TreeNode{
    Val:   8,
    Left:  nil,
    Right: nil,
   },
  },
 }
 l := &TreeNode{
  Val:   5,
  Left:  nil,
  Right: nil,
 }
 r := &TreeNode{
  Val:   1,
  Left:  nil,
  Right: nil,
 }
func lowestCommonAncestor(root, p, q *TreeNode) *TreeNode {
 if root == nil {
  return nil
 }
 if root.Val == p.Val || root.Val == q.Val {
  return root
 }
 left := lowestCommonAncestor(root.Left, p, q)
 right := lowestCommonAncestor(root.Right, p, q)
 if left != nil && right != nil {
  return root
 }
 if left == nil {
  return right
 }
 return left
}

                

                    BST查找最近公共父节点

BST的话,是左侧全部小于跟节点,右侧全部大于跟节点,基于这个特点,我们可以看q、p是比root大还是小,还是有大有小

func lowestCommonAncestor(root, p, q *TreeNode) *TreeNode {
 if root == nil {
  return nil
 }

 if (root.Val - p.Val) * (root.Val - q.Val) < 0{
  return root
 }
 if p.Val < root.Val{
  return lowestCommonAncestor(root.Left, p, q)
 }
 if p.Val > root.Val{
  return lowestCommonAncestor(root.Right, p, q)
 }
 return nil
}


码虫甲
专注后端Java技术、大数据领域、数据结构与算法、面经分享等优质内容输出
4篇原创内容
Official Account