题目(leecode T236):
给定一个二叉树, 找到该树中两个指定节点的最近公共祖先。
百度百科中最近公共祖先的定义为:“对于有根树 T 的两个节点 p、q,最近公共祖先表示为一个节点 x,满足 x 是 p、q 的祖先且 x 的深度尽可能大(一个节点也可以是它自己的祖先)。
方法:本题要找到给定两个节点的公共祖先,我们要自下而上的找,后序遍历正好可以满足这个需求。因此我们需要采用基于后序遍历的递归方式。分析递归三部曲
1:传入参数与返回值:传入参数有整个树的根节点,需要查找公共在祖先的两个节点。返回的是找到的公共节点
2:终止条件:首先遇到空节点肯定是要返回的,其次是如果根节点是p或q节点的话,那么它本身肯定就是公共祖先节点了(因为祖先节点可以是他本身)
3:单层处理逻辑:我们递归得到左子树与右子树的递归结果后,需要对其返回值进行判断,如果左右子树都不为空,此时的root节点就是最近公共祖先,如果一个为空一个不为空,那么不为空的就是公共祖先,如果都为空说明没有,就是返回NULL。
题解:
class Solution {
public:
TreeNode* lowestCommonAncestor(TreeNode* root, TreeNode* p, TreeNode* q) {
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 root;
if(left != NULL && right == NULL) return left;
else if(left == NULL && right != NULL) return right;
else return NULL;
}
};