题目描述
给定一个二叉树, 找到该树中两个指定节点的最近公共祖先。树的输入是按照层次来的,这样就是保证
i
∗
2
i*2
i∗2位置是左边儿子,
i
∗
2
+
1
i*2+1
i∗2+1是右儿子。
示例 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。因为根据定义最近公共祖先节点可以为节点本身。
tips:
- 所有节点的值都是唯一的。
- p、q 为不同节点且均存在于给定的二叉树中。
题解:
由tips中的条件得知,我们可以利用递归,dfs(root,p,q)代表以root为根节点,p,q的lca。那么递归的终止条件是:如果root为空,那么就直接返回空,如果其中有一个它就等于root,那么直接返回root。如果不满足递归的终止条件,递归左右儿子。
/**
* Definition for a binary tree node.
* function TreeNode(val) {
* this.val = val;
* this.left = this.right = null;
* }
*/
/**
* @param {TreeNode} root
* @param {TreeNode} p
* @param {TreeNode} q
* @return {TreeNode}
*/
var lowestCommonAncestor = function(root, p, q) {
if(root == null) return null
else if(p == root || q == root) return root
let ls = new TreeNode()
let rs = new TreeNode()
ls = lowestCommonAncestor(root.left, p, q)
rs = lowestCommonAncestor(root.right, p, q)
if(ls && rs) return root
else if(ls) return ls
else if(rs) return rs
};