给定一个二叉树, 找到该树中两个指定节点的最近公共祖先。
百度百科中最近公共祖先的定义为:“对于有根树 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
提示:
- 树中节点数目在范围
[2, 105]
内。 -109 <= Node.val <= 109
- 所有
Node.val
互不相同
。 p != q
p
和q
均存在于给定的二叉树中。
题解:
/**
* @description: 深度优先 TC:O(n) SC:O(n)
* @author: JunLiangWang
* @param {*} root 给定树的根节点
* @param {*} p 指定节点p
* @param {*} q 指定节点q
* @return {*}
*/
function dfs(root,p,q){
/**
* 本方案使用深度优先的方式遍历树的节点,
* 由于树中节点值是不重复的,因此我们仅需
* 在遍历中记录已找到的与p/q节点相等的节点
* 的数量即可,当数量达到2,即证明已找到了
* p/q节点,然后当前根节点即为其最近的公共
* 根。
*/
// 记录公共根节点
let rootNode=null
/**
* @description: 利用递归实现深度优先
* @author: JunLiangWang
* @param {*} root 当前子树根节点
* @return {*}
*/
function recursion(root){
// 如果根节点为空返回0
if(!root)return 0;
// 获得左右子树遍历的结果,并相加
let count=recursion(root.left)+recursion(root.right)
// 比较当前根节点的值是否与p或q相等,如果相等count+1
if(root.val==p.val||root.val==q.val)count++;
// 如果count==2,证明已找到了
// p/q节点,然后当前根节点即为
// 其最近的公共根。
if(count==2){
// 记录公共节点
rootNode=root
// count+1,防止再次进入
count++;
}
return count
}
// 执行递归
recursion(root)
// 返回结果
return rootNode
}