https://leetcode-cn.com/problems/lowest-common-ancestor-of-a-binary-tree/
給定一個二叉樹, 找到該樹中兩個指定節點的最近公共祖先。
百度百科中最近公共祖先的定義爲:“對於有根樹 T 的兩個結點 p、q,最近公共祖先表示爲一個結點 x,滿足 x 是 p、q 的祖先且 x 的深度儘可能大(一個節點也可以是它自己的祖先)。”
例如,給定如下二叉樹: root = [3,5,1,6,2,0,8,null,null,7,4]
示例 1:
輸入: root = [3,5,1,6,2,0,8,null,null,7,4], p = 5, q = 1
輸出: 3
解釋: 節點 5 和節點 1 的最近公共祖先是節點 3。
示例 2:
輸入: root = [3,5,1,6,2,0,8,null,null,7,4], p = 5, q = 4
輸出: 5
解釋: 節點 5 和節點 4 的最近公共祖先是節點 5。因爲根據定義最近公共祖先節點可以爲節點本身。
說明:
所有節點的值都是唯一的。
p、q 爲不同節點且均存在於給定的二叉樹中。
# Definition for a binary tree node.
# class TreeNode(object):
# def __init__(self, x):
# self.val = x
# self.left = None
# self.right = None
class Solution(object):
def lowestCommonAncestor(self, root, p, q):
"""
:type root: TreeNode
:type p: TreeNode
:type q: TreeNode
:rtype: TreeNode
"""
m = {}
def dfs(root):
if root:
if root.left:
m[root.left] = root
if root.right:
m[root.right] = root
dfs(root.left)
dfs(root.right)
dfs(root)
l1, l2 = p, q
while l1 != l2:
l1 = m.get(l1, None) if m.get(l1, None) is not None else q
l2 = m.get(l2, None) if m.get(l2, None) is not None else p
return l1
思路:
由於每個節點只有唯一一個父節點,我們可以使用到字典的value-key的形式(節點-父節點)字典中預置根節點的父節點爲None。字典建立完成後,二叉樹就可以看成一個所有節點都將最終指向根節點的鏈表了。於是在二叉樹中尋找兩個節點的最小公共節點就相當於,在一個鏈表中尋找他們相遇的節點。
作者:mian-mian-sir
鏈接:https://leetcode-cn.com/problems/lowest-common-ancestor-of-a-binary-tree/solution/shi-yong-zi-dian-cun-chu-shuang-qin-jie-dian-by-mi/
上面代碼中dfs爲深度遍歷二叉樹,建立子節點指向根節點的字典。其中根節點root我沒建立,後面可以默認None,然後重點來了,將兩個子節點分別同時尋找它的根節點,如果兩個節點能夠相等,說明已經找到根節點,如果一個節點發現沒有父節點,說明是上面圖示的情況,即一個節點離LCA近,一個離LCA遠。那麼我們開始遍歷每個節點的根節點,當l2已經到達根節點root時,l1還沒有到,並且此刻l1和l2的距離和一開始l1,l2距離是一樣的,那麼此時將l1初始的位置賦值給l2,繼續遍歷;當l1到達根節點時,l2達到到達與初始l2相同的距離(注意不是相同的位置,是距離根節點一樣的距離),此時將l2的初始位置賦值給l1,那麼此時l1和l2距離LCA就是一樣的距離了,那麼現在只要兩個節點繼續遍歷下去就肯定能找到LCA。
此時將另一個節點賦值給該節點,然後繼續尋找父節點;之後另一個節點也會發現沒有父節點,則將