二叉树的最近公共祖先｜Go主题月

深海001
1,293 阅读3分钟

【Golang主题学习月】 刷题比玩游戏好多了，成就感越来越强，每天坚持刷一道题，每天锻炼30分钟，等8块腹肌,等大厂offer.

😄

 \color{red}{~}

我相信，如果在面试中遇到此题，逻辑清晰、正确表达出来、手撕

应该会超过一部分的面试者。

对树不熟悉的朋友，可以看看前面的基础训练题哦！

leecode 236. 二叉树的最近公共祖先

给定一个二叉树, 找到该树中两个指定节点的最近公共祖先。

百度百科中最近公共祖先的定义为：“对于有根树 T 的两个节点 p、q，最近公共祖先表示为一个节点 x，满足 x 是 p、q 的祖先且 x 的深度尽可能大（一个节点也可以是它自己的祖先）。”

 

示例 1：

图片.png

输入：root = [3,5,1,6,2,0,8,null,null,7,4], p = 5, q = 1

输出：3

解释：节点 5 和节点 1 的最近公共祖先是节点 3 。

示例 2：

图片.png

输入：root = [3,5,1,6,2,0,8,null,null,7,4], p = 5, q = 4

输出：5

解释：节点 5 和节点 4 的最近公共祖先是节点 5 。因为根据定义最近公共祖先节点可以为节点本身。

参考代码

定义一颗树

/**
 * Definition for a binary tree node.
 * type TreeNode struct {
 *     Val int          // 根
 *     Left *TreeNode   //左节点
 *     Right *TreeNode  //右节点
 * }
 */

GO语言版 递归

  1. 要知道这是一棵树，只是通过数组的形式表现出来的

图片.png

简单的用这颗树来举例

关键点 ：

  1. 递归出口

  2. 去该节点的左子树上找

  3. 去该节点的右子树上找

  4. 左右均有，说明该节点即为最近公共祖先

  5. 左子树上没有，说明在右子树上

  6. 右子树上没有，说明在左子树上

func lowestCommonAncestor(root, p, q *TreeNode) *TreeNode {
    if root == nil {  // 1. 
        return nil  
    }
    if root.Val == p.Val || root.Val == q.Val {
        return root
    }
    left := lowestCommonAncestor(root.Left, p, q)  // 2. 
    right := lowestCommonAncestor(root.Right, p, q)  // 3. 
    if left != nil && right != nil {
        return root   // 4. 
    }
    if left == nil {   // 5. 
        return right
    }
    return left  // 5. 
}

迭代版

核心和递归基本上差不多的。

 func lowestCommonAncestor(root, p, q *TreeNode) *TreeNode {
    parent := map[int]*TreeNode{}
    visited := map[int]bool{}

    var dfs func(*TreeNode)
    dfs = func(r *TreeNode) {
        if r == nil {
            return
        }
        if r.Left != nil {
            parent[r.Left.Val] = r
            dfs(r.Left)
        }
        if r.Right != nil {
            parent[r.Right.Val] = r
            dfs(r.Right)
        }
    }
    dfs(root)

    for p != nil {
        visited[p.Val] = true
        p = parent[p.Val]
    }
    for q != nil {
        if visited[q.Val] {
            return q
        }
        q = parent[q.Val]
    }

    return nil
}

真心感谢帅逼靓女们能看到这里，如果这个文章写得还不错，觉得有点东西的话

求点赞👍 求关注❤️ 求分享👥 对8块腹肌的我来说真的 非常有用！！！

如果本篇博客有任何错误，请批评指教，不胜感激 ！❤️❤️❤️❤️

avatar
文章
阅读
粉丝