给定一棵二叉搜索树, 找到该树中两个指定节点的最近公共祖先。
百度百科中最近公共祖先的定义: “对于有根树T的两个结点u、v,最近公共祖先表示一个结点x,满足x是u、v的祖先且x的深度尽可能大。”(一个节点也可以是它自己的祖先)
_______6______ / ___2__ ___8__ / / 0 _4 7 9 / 3 5
例如, 节点 2
和 8
的最近公共祖先是 6
。再举个例子,节点 2
和 4
的最近公共祖先是 2
, 因为根据定义最近公共祖先节点可以为指定节点自身。
思路:
- 如果root的值和p,q的值相等,则返回root
-
如果root的值在p,q的值的中间,那么该root就是p,q的公共祖先
-
如果root的值大于p和q的值,则在root的左子树中寻找p,q的公共祖先
-
如果root的值大于p和q的值,则在root的右子树中寻找p,q的公共祖先
1 class Solution { 2 public: 3 TreeNode* lowestCommonAncestor(TreeNode* root, TreeNode* p, TreeNode* q) { 4 if(root->val == p->val || root->val == q->val) return root; 5 if((root->val > p->val && root->val < q->val)||(root->val < p->val && root->val > q->val)) return root; 6 if(root->val < p->val && root->val < q->val) return lowestCommonAncestor(root->right, p, q); 7 if(root->val > p->val && root->val > q->val) return lowestCommonAncestor(root->left, p,q); 8 } 9 };