设计并实现一个算法,找出二叉树中某两个节点的第一个共同祖先。不得将其他的节点存储在另外的数据结构中。注意:这不一定是二叉搜索树。
例如,给定如下二叉树: root = [3,5,1,6,2,0,8,null,null,7,4]
3
/
5 1
/ \ /
6 2 0 8
/
7 4
递归中止条件 如果当前查找的节点是空的那么函数结束, 或者根节点等于p,q中任意一个节点,那么也相当于找到我们所找的公共祖先(根据定义最近公共祖先节点可以为节点本身)
分情况讨论, left为在左子树中寻找公共祖先或p,q节点的结果, right为在右子树中寻找公共祖先或p,q节点的结果 2.1 如果left与right都不为空,证明在左右子树各找到了p,q中的一个,所以当前节点肯定是公共祖先,返回即可 2.2 如果left不为空right为空,证明要找的祖先在左子树中,返回左子树的递归结果即可 2.3 如果left为空right不为空,同上
TreeNode* lowestCommonAncestor(TreeNode* root, TreeNode* p, TreeNode* q) {
if(root == NULL)
return NULL;
if(root == p || root == q)
return root;
TreeNode *left = lowestCommonAncestor(root->left, p, q);
TreeNode *right = lowestCommonAncestor(root->right, p, q);
if(left != NULL && right != NULL)
return root;
else if(left != NULL)
return left;
else
return right;
}