LeetCode235. 二叉搜尋樹的最近公共祖先
阿新 • • 發佈:2020-08-21
二叉搜尋樹的性質:(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; } };