题目大意:在二叉搜索树中找出两节点的最近祖先节点
分析:首先明确概念,只要B节点位于A节点的子树中,A即为B的祖先节点(自己也是自己的祖先节点);“最近”是指两节点的祖先节点中深度最大的一个。两节点的最近祖先节点特征:值位于两节点之间(包含端点值)。因此从根节点递归遍历,当前节点大于两节点时,继续遍历左子树;当前节点小于两节点时,继续遍历右子树。
代码:
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->val > p->val && root->val > q->val) return lowestCommonAncestor(root->left, p, q);
if(root->val < p->val && root->val < q->val) return lowestCommonAncestor(root->right, p, q);
return root;
}
};