235. 二叉搜尋樹的最近公共祖先
阿新 • • 發佈：2020-08-17
235. 二叉搜尋樹的最近公共祖先
給定一個二叉搜尋樹, 找到該樹中兩個指定節點的最近公共祖先。 百度百科中最近公共祖先的定義為：“對於有根樹 T 的兩個結點 p、q，最近公共祖先表示為一個結點 x，滿足 x 是 p、q 的祖先且 x 的深度儘可能大（一個節點也可以是它自己的祖先）。” 例如，給定如下二叉搜尋樹: root = [6,2,8,0,4,7,9,null,null,3,5] 示例 1: 輸入: root = [6,2,8,0,4,7,9,null,null,3,5], p = 2, q = 8輸出: 6
解釋: 節點 2 和節點 8 的最近公共祖先是 6。
輸出: 2
解釋: 節點 2 和節點 4 的最近公共祖先是 2, 因為根據定義最近公共祖先節點可以為節點本身。 說明:
所有節點的值都是唯一的。
p、q 為不同節點且均存在於給定的二叉搜尋樹中。 思路：回溯 程式碼：
/** * 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(!root||root==p||root==q) return root; 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; } };
思路：迭代法
我們只需要找到分割點就可以了。這個分割點就是能讓節點 p 和節點 q 不能在同一顆子樹上的那個節點，或者是節點 p 和節點 q 中的一個，這種情況下其中一個節點是另一個節點的父親節點。時間複雜度 o(N),空間複雜度 o(1)。 程式碼：/** * 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) { TreeNode *node = root; while(node) { if(p->val>node->val&&q->val>node->val) node = node->right; else if(p->val<node->val&&q->val<node->val) node = node->left; else break; } return node; } };