[Leetcode]36.二叉树的最近公共祖先

题目:

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

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

示例 1:

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

示例 2:

输入: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

 

思想:深度搜索,找到两个路径,遍历找到公共节点。

func lowestCommonAncestor(root, p, q *TreeNode) *TreeNode {

	var path []*TreeNode
	var path_p []*TreeNode
	var path_q []*TreeNode
	var res *TreeNode
	var dfs func(root *TreeNode)
	dfs = func(root *TreeNode){
		if root!=nil{
            path = append(path,root)
            if root == p{
                path_p =make([]*TreeNode,len(path))
                copy(path_p,path)
            }
            if root == q{
                path_q =make([]*TreeNode,len(path))
                copy(path_q,path)
            }
          
            dfs(root.Left)
            dfs(root.Right)
            path = path[:len(path)-1]
        }
	}
	dfs(root)
	for i:=0;i<len(path_q)&&i<len(path_p)&&path_p[i]==path_q[i];i++{
		res = path_q[i]
	}
	return res
}               

题目来源:https://leetcode-cn.com/problems/lowest-common-ancestor-of-a-binary-tree

posted @ 2021-11-21 10:48  梦想是能睡八小时的猪  阅读(32)  评论(0)    收藏  举报