leetcode 235. 二叉搜索树的最近公共祖先

给定一棵二叉搜索树, 找到该树中两个指定节点的最近公共祖先。

百度百科中最近公共祖先的定义: “对于有根树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 };

猜你喜欢

转载自www.cnblogs.com/mr-stn/p/8969023.html