Leetcode 1816 Solution

This article provides solution to leetcode question 1816 (lowest-common-ancestor-of-a-binary-tree-iv)
Categories:
Tags:
Access this page by simply typing in "lcs 1816" in your browser address bar if you have bunnylol configured.

https://leetcode.com/problems/lowest-common-ancestor-of-a-binary-tree-iv

Solution

# 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: 'TreeNode', nodes: 'List[TreeNode]') -> 'TreeNode':
        nodes = set([node.val for node in nodes])

        ans = None
        def dfs(node):
            nonlocal ans

            if ans:
                return 0

            if not node:
                return 0

            cnt = int(node.val in nodes)
            cnt += dfs(node.left)
            cnt += dfs(node.right)

            if ans == None and cnt == len(nodes):
                ans = node
                return 0

            return cnt

        dfs(root)
        return ans