1. 程式人生 > >leetcode 235. 二叉搜尋樹的最近公共祖先(Python)[Easy]

leetcode 235. 二叉搜尋樹的最近公共祖先(Python)[Easy]

題目:

給定一個二叉搜尋樹, 找到該樹中兩個指定節點的最近公共祖先。

百度百科中最近公共祖先的定義為:“對於有根樹 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