由于之前做过leetcode235的二叉搜索树的寻找最近公共祖先,由于二叉搜索树的天然定义,所以很好解。
而这是一个普通的二叉树,当然,面对二叉树问题首先想到的就是可否用递归来做
class Solution {
public:
TreeNode* lowestCommonAncestor(TreeNode* root, TreeNode* p, TreeNode* q) {
assert( p != NULL && q != NULL );
if( root == NULL || 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 NULL;
else if(left != NULL && right != NULL) return root; //两个节点分布在两侧
else return left == NULL ? right : left;
}
};
这里递归找左子树和右子树。如果两个树都找到了,即p,q分隔在树的两侧。即最近公共节点就为原始root。
如果一侧为空,则两个节点都聚集在树的一侧,这样,先找到的节点就是最近公共节点
如果都没找到,即树中两个节点不存在,返回NULL即可