235.[Leetcode]Lowest Common Ancestor of a Binary Search Tree

class Solution {
public:
TreeNode* lowestCommonAncestor(TreeNode* root, TreeNode* p, TreeNode* q) {
TreeNode* temp = root;
//首先如果根节点和其中一个相同，那么肯定返回根节点
if(temp == q || temp == p) {
return root;
}
while (temp!=NULL) {
if(IsNode(temp->left,p) && IsNode(temp->left,q)) temp = temp->left; // 如果p，q都是temp的左子节点的子节点，则关注点移到左边，因为右边已经没有可能了
else if(IsNode(temp->right,p) && IsNode(temp->right,q)) temp = temp->right; // 同理
else break; // 如果两边都不是，那么肯定是本身是一个LCA
}
return temp;
}

// isNode 用来判断是否 n 是 root 的子节点
bool IsNode(TreeNode* root,TreeNode* n) {
if(root == n) return true;
if(root == NULL) return false;
else return IsNode(root->right,n) || IsNode(root->left,n);
}
};