Create find-closest-node-to-given-two-nodes.py

Author: kamyu104

Python/find-closest-node-to-given-two-nodes.py

@@ -0,0 +1,26 @@
+# Time:  O(n)
+# Space: O(n)
+
+# hash table
+class Solution(object):
+    def closestMeetingNode(self, edges, node1, node2):
+        """
+        :type edges: List[int]
+        :type node1: int
+        :type node2: int
+        :rtype: int
+        """
+        def dfs(node):
+            lookup = {}
+            i = 0
+            while node != -1:
+                if node in lookup:
+                    break
+                lookup[node] = i
+                i += 1
+                node = edges[node]
+            return lookup
+        
+        lookup1, lookup2 = dfs(node1), dfs(node2)
+        intersect = set(lookup1.iterkeys())&set(lookup2.iterkeys())
+        return min(intersect, key=lambda x: (max(lookup1[x], lookup2[x]), x)) if intersect else -1