leetcode 235. 二叉搜尋樹的最近公共祖先(Python)[Easy]
阿新 • • 發佈:2018-11-10
題目:
給定一個二叉搜尋樹, 找到該樹中兩個指定節點的最近公共祖先。
百度百科中最近公共祖先的定義為:“對於有根樹 T 的兩個結點 p、q,最近公共祖先表示為一個結點 x,滿足 x 是 p、q 的祖先且 x 的深度儘可能大(一個節點也可以是它自己的祖先)。”
例如,給定如下二叉搜尋樹: root = [6,2,8,0,4,7,9,null,null,3,5]
_______6______ / \ ___2__ ___8__ / \ / \ 0 _4 7 9 / \ 3 5
示例 1:
輸入: root = [6,2,8,0,4,7,9,null,null,3,5], p = 2, q = 8 輸出: 6 解釋: 節點2
和節點8
的最近公共祖先是6。
示例 2:
輸入: root = [6,2,8,0,4,7,9,null,null,3,5], p = 2, q = 4 輸出: 2 解釋: 節點2
和節點4
的最近公共祖先是2
, 因為根據定義最近公共祖先節點可以為節點本身。
說明:
- 所有節點的值都是唯一的。
- p、q 為不同節點且均存在於給定的二叉搜尋樹中。
思路:
題目中給出是一個二叉搜尋樹。二叉搜尋樹中根節點總是比左子樹上的每一個數大,總是比右子樹每一個數小。因此p和q的公共祖先的值應該介於左子樹和右子樹之間。從根節點開始遍歷,如果根節點的數值均小於p和q的數值,那麼代表p和q 的祖先在根節點的右子樹中。如果根節點的數值均大於p和q的數值,那麼代表p和q 的祖先在根節點的左子樹中。若不滿足上述,就代表祖先節點就是pq中靠近根節點的一個。
程式碼:
# Definition for a binary tree node. # class TreeNode: # def __init__(self, x): # self.val = x # self.left = None # self.right = None class Solution: def lowestCommonAncestor(self, root, p, q): """ :type root: TreeNode :type p: TreeNode :type q: TreeNode :rtype: TreeNode """ if root == None: return None if (root.val>p.val and root.val<q.val) or (root.val<p.val and root.val>q.val): return root if root.val > p.val and root.val >q.val: return self.lowestCommonAncestor(root.left,p,q) if root.val < p.val and root.val <q.val: return self.lowestCommonAncestor(root.right,p,q) return root