时间复杂度:O(n)
解题思路
还是在递归的过程中解决问题。
由于涉及到父子结点的关系判断,所以考虑用后序遍历的方法递归。
如果遍历的结点是我们要寻找祖先的结点,那么就应该通知给它的父节点,让它的父节点知道左孩子或是右孩子是一个要寻找公共祖先的节点。然后父节点对左右孩子返回给它的节点做比较,如果都是要寻找公共祖先的节点,就说明此时的父节点就是最近公共祖先;如果其中一个孩子是要寻找公共祖先的节点,那么就返回它,告诉父节点的父节点它的孙子有一个要寻找公共祖先的节点;如果两个孩子都不是要寻找公共祖先的节点,那么就正常返回nil。
AC代码
/**
* Definition for a binary tree node.
* type TreeNode struct {
* Val int
* Left *TreeNode
* Right *TreeNode
* }
*/
func lowestCommonAncestor(root, p, q *TreeNode) *TreeNode {
if root==nil||root==p||root==q{
return root
}
left:=lowestCommonAncestor(root.Left,p,q)
right:=lowestCommonAncestor(root.Right,p,q)
if left!=nil{
if right!=nil{
return root//left和right都不空说明root就是最近公共祖先
}
return left//root是其中一个结点的祖先
}
return right
}
感悟
在比较左右孩子的时候遇到了些麻烦,没想好应该怎么处理,但是当举了一个具体的例子后,就很容易想通应该怎么做了。