初步的思路是把长度超过2的node做2分,划分到长度为1 或者2的node sublist,这样就可以拿到LCA里处理。但是这样做会在第54(/57)个TC处TLE。先把这个写法留下,之后写可以全部过的版本。
class Solution:
def lowestCommonAncestor(self, root: 'TreeNode', nodes: 'List[TreeNode]') -> 'TreeNode':
if len(nodes) == 1:return nodes[0]
if len(nodes) == 2:return self.findLCA(root, nodes[0], nodes[1])
if len(nodes) > 2:
sep = len(nodes)//2
nodelist1 = nodes[:sep]
nodelist2 = nodes[sep:]
ans1 = self.lowestCommonAncestor(root, nodelist1)
ans2 = self.lowestCommonAncestor(root, nodelist2)
return self.findLCA(root, ans1, ans2)
def findLCA(self, root, node1, node2):
if not root:return None
if root.val == node1.val or root.val == node2.val:return root
leftpath = self.findLCA(root.left, node1, node2)
rightpath = self.findLCA(root.right, node1, node2)
if leftpath and rightpath:return root
return leftpath if leftpath else rightpath
接下来是全部通过的版本,其实只要这样思考,root在nodes中,那一定是nodes, 而不在,则尝试root.left 和root.right看是否在nodes中,在就返回,不在,则继续node的left和right的left和right孙节点是否在nodes中,直到找到。
class Solution:
def lowestCommonAncestor(self, root: 'TreeNode', nodes):
if not root:return None
if root in nodes:return root
left = self.lowestCommonAncestor(root.left, nodes)
right = self.lowestCommonAncestor(root.right, nodes)
if left and right:return root
if left:return left
if right:return right
return None