参考236. Lowest Common Ancestor of a Binary Tree
但因为这题是BST,所以有别的方法
以下情况:
- 当p,q都在root左侧,那么p,q都在root左子树。
- 当p,q都在root右侧,那么p,q都在root右子树。
- 当第一次产生没在同一个子树时,这时候的root就是解
- p或q为root,剩下的在一边的树
- p,q在root两侧
迭代
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24
|
class Solution { public: TreeNode* lowestCommonAncestor(TreeNode* root, TreeNode* p, TreeNode* q) { while(root) { if(root->val > p->val && root->val > q->val) root = root->left; else if(root->val < p->val && root->val < q->val) root = root->right; else return root; } return nullptr; } };
|
递归
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19
|
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); else if(root->val < p->val && root->val < q->val) return lowestCommonAncestor(root->right, p, q); return root; } };
|