题目描述:
给定一棵二叉树以及这棵树上的两个节点 o1 和 o2,请找到 o1 和 o2 的最近公共祖先节点。
示例1
输入
[3,5,1,6,2,0,8,#,#,7,4],5,1
返回值
3
分析:
有四种情况:
(1)当o1或o2任意一个为根节点时,那么两个节点的最近公共祖先就是根节点
(2)当o1和o2一个在根节点的左树一个在右树时,最近公共祖先为根节点
(3)当o1和o2都在左树时,最近公共祖先是左树节点
(4)当o1和o2都在右树时,最近公共祖先是右树节点
class Solution {
public:
int lowestCommonAncestor(TreeNode* root, int o1, int o2) {
// write code here
if(rootNULL)
return 0;
if(root->valo1||root->val==o2)//当根节点为o1或o2时,其最近的公共祖先就是o1或o2
{
return root->val;
} //用left和right分别接收root的左右子树
int left=lowestCommonAncestor(root->left,o1,o2);
int right=lowestCommonAncestor(root->right,o1,o2);
//判断左右子树是否为空
if(left!=NULL&&right!=NULL)//root节点的左右子树都不为空
{
return root->val;
}
if(left!=NULL)//左子树不为空,右子树为空
return left;
if(right!=NULL) //左子树为空,右子树不为空
return right;
return NULL;//说明在root的左右子树中没有找到o1和o2
}
};