可以根据二叉搜索树的特点来进行解题:
- 二叉搜索树左边节点的值小于根节点
- 右边节点的值大于根节点。
所以左右都小于根节点的值说明在左子树上
左右的值大于根节点的值就代表值在右子树上
介于两者中间则输入的本身就是根节点,即为最近公共祖先。
一、遍历
1 /** 2 * Definition for a binary tree node. 3 * struct TreeNode { 4 * int val; 5 * TreeNode *left; 6 * TreeNode *right; 7 * TreeNode(int x) : val(x), left(NULL), right(NULL) {} 8 * }; 9 */ 10 class Solution { 11 public: 12 TreeNode* lowestCommonAncestor(TreeNode* root, TreeNode* p, TreeNode* q) { 13 if(root==NULL) return NULL; 14 bool flag=true; 15 while(flag){ 16 //两个值都小于根节点的值,则在左子树上面 17 if(root->val > p->val && root->val > q->val) root=root->left; 18 //都大于根节点值,则在右子树上面 19 else if(root->val < p->val && root->val < q->val) root=root->right; 20 else 21 flag = false;//要么其中一个节点=根节点,要么这两个节点分别在两颗子树上,那么直接返回根节点就行了 22 } 23 return root; 24 } 25 };
二、递归
1 /** 2 * Definition for a binary tree node. 3 * struct TreeNode { 4 * int val; 5 * TreeNode *left; 6 * TreeNode *right; 7 * TreeNode(int x) : val(x), left(NULL), right(NULL) {} 8 * }; 9 */ 10 class Solution { 11 public: 12 TreeNode* lowestCommonAncestor(TreeNode* root, TreeNode* p, TreeNode* q) { 13 //首先先看是否是空树 14 if(root==NULL) return NULL; 15 //递归的终止条件:如果根节点就是p或者q,那个就返回根节点啊 16 if(root==p || root==q) return root; 17 //开始递归,递归的逻辑是从倒数第二步开始推 18 //找节点是q或p的 19 TreeNode* left = lowestCommonAncestor(root->left,p,q); 20 TreeNode* right = lowestCommonAncestor(root->right,p,q); 21 //倒数第二步来看,左右都是NULL,说明没有子节点等于p或q,那么就直接返回root就行了 22 //但是实际上已经说了p、q是在二叉树里面的,所以这种情况可以抛弃 23 if(left!=NULL && right !=NULL) return root; 24 //左非空,右空,那么这个left是最近的了 25 else if(left != NULL) return left; 26 else if(right != NULL) return right; 27 return NULL; 28 } 29 };
原文地址:https://www.cnblogs.com/pacino12134/p/11073726.html
时间: 2024-11-08 09:56:12