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);
    }
};