1. 程式人生 > 實用技巧 >LeetCode235. 二叉搜尋樹的最近公共祖先

LeetCode235. 二叉搜尋樹的最近公共祖先

二叉搜尋樹的性質:(1)root為空;(2)或者root左子樹中所有元素都小於root->val,root右子樹中所有元素都大於root->val,
且root的左右子樹也是二叉搜尋樹。

所以要找兩個節點的最近公共祖先,有三種情況:

(1)p,q都在root的左子樹中,則我們到root的左子樹去尋找他們的祖先(因為需要找最近的祖先);

(2)p,q都在root的右子樹中,則我們到root的右子樹去尋找他們的祖先;

(3)其他情況:可能p,q分別在root的左右子樹或者反過來,也有可能p和q其中之一就是root,這些情況下,p和q的最近公共祖先都是root。

程式碼:

/**
 * Definition for a binary tree node.
 * 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(p -> val < root -> val && q -> val < root -> val) {
            return lowestCommonAncestor(root -> left, p, q);
        }
        if(p -> val > root -> val && q -> val > root -> val) {
            return lowestCommonAncestor(root -> right, p, q);
        }
        return root;
    }
};