题目描述:
给定一个二叉树, 找到该树中两个指定节点的最近公共祖先。
百度百科中最近公共祖先的定义为:对于有根树 T 的两个结点 p、q,最近公共祖先表示为一个结点 x,满足 x 是 p、q 的祖先且 x 的深度尽可能大(一个节点也可以是它自己的祖先。
分析题目:
1.输入是什么?输入是一个二叉树和此树中两个指定节点p和q。
2.输出是什么?输出是一个满足所谓最近公共祖先要求的树节点。
3.采用什么算法或数据结构?因为是二叉树,首先想到能不能递归求解!
4.函数名直接用“最近公共祖先”的英译:lowestCommonAncestor。
至此,你应该可以写出如下函数声明:
TreeNode* lowestCommonAncestor(TreeNode* root, TreeNode* p, TreeNode* q)
思路:
拟采用递归算法,具体如何个递归套路呢?是先序,中序,后续还是BFS?
第一步:递归结束条件。
当根节点时NULL的时候,递归结束,当根节点等于p或者q的时候,递归也可以结束。
第二步:我们递归的去做一件什么事可以解决这个问题呢?
根据第一步的递归返回条件,我们可以推断出我们需要递归的去在左右子树中查找是否存在p节点或者q节点,如果存在,那递归返回的那个值就是我们的结果!
而查找左右子树上是否有p结点或者q结点,一共有4种情况:
第1种情况:左子树和右子树均找没有p结点或者q结点;(这里特别需要注意,虽然题目上说了p结点和q结点必定都存在,但是递归的时候必须把所有情况都考虑进去,因为题目给的条件是针对于整棵树,而递归会到局部,不一定都满足整体条件)
第2种情况:左子树上能找到,但是右子树上找不到,此时就应当直接返回左子树的查找结果;
第3种情况:右子树上能找到,但是左子树上找不到,此时就应当直接返回右子树的查找结果;
第4种情况:左右子树上均能找到,说明此时的p结点和q结点分居root结点两侧,此时就应当直接返回root结点了。
解答:
/**
* Definition for a binary tree node.
* struct TreeNode {
* int val;
* TreeNode *left;
* TreeNode *right;
* TreeNode(int x) : val(x), left(NULL), right(NULL) {}
* };
*/
class Solution {
public:
TreeNode* lowestCommonAncestor(TreeNode* root, TreeNode* p, TreeNode* q) {
if(root == NULL)
{
return NULL;
}
//先找到p或者q就返回
if(root == p || root == q)
{
return root;
}
//在左子树找p或q,假如p和q都在左子树中,返回的那个值就是祖先
TreeNode *left = lowestCommonAncestor(root->left, p, q);
//在右子树找p或者q,假如p和q都在右子树中,返回的那个值就是祖先
TreeNode *right = lowestCommonAncestor(root->right, p, q);
if(left == NULL && right == NULL)
{
return NULL;
}
else if(left == NULL && right != NULL)
{
return right;
}
else if(right == NULL && left != NULL)
{
return left;
}
//p和q一个在左子树,一个在右子树
return root;
}
};