题目:
给定一棵二叉树,找到两个节点的最近公共父节点(LCA)。
最近公共祖先是两个节点的公共的祖先节点且具有最大深度。
注意事项:
假设给出的两个节点都在树中存在。
样例:
思路:
查找两个node的最早的公共祖先,分三种情况:
- 如果两个node在root的两边,那么最早的公共祖先就是root。
- 如果两个node在root的左边,那么把root的左子树作为root,再递归。
- 如果两个node在root的右边,那么把root的右子树作为root,再递归。
自底向上的遍历二叉树(深度优先遍历),一旦找到了两个节点其中的一个,就将这个几点返回给上一层,上一层节点通过判断其左右子树中是否恰好包含n1和n2两个节点,如果是这样的话,对应的上一层节点肯定是所求的LCA;若果不是的话,将包括两个节点中任意一个的较低的节点返回给上一层,否则返回NULL。
参考答案
/**
* Definition of TreeNode:
* class TreeNode {
* public:
* int val;
* TreeNode *left, *right;
* TreeNode(int val) {
* this->val = val;
* this->left = this->right = NULL;
* }
* }
*/
class Solution {
public:
/*
* @param root: The root of the binary search tree.
* @param A: A TreeNode in a Binary.
* @param B: A TreeNode in a Binary.
* @return: Return the least common ancestor(LCA) of the two nodes.
*/
TreeNode * lowestCommonAncestor(TreeNode * root, TreeNode * A, TreeNode * B) {
// write your code here
//如果当前节点为空,或者与目标节点中的一个相同,则返回该节点
if(root == NULL) return NULL;
if(root==A || root==B) return root;
//递归寻找A B在左右子树的位置
TreeNode* left = lowestCommonAncestor(root->left,A,B);
TreeNode* right = lowestCommonAncestor(root->right,A,B);
//如果A B分别位于root的两侧,则root是他们的LCA,否则是左子树或者右子树
if(left&&right) return root;
return left?left:right;
}
};
**粗体** _斜体_ [链接](http://example.com) `代码` - 列表 > 引用
。你还可以使用@
来通知其他用户。