leetcode.com 399. Evaluate Division

문제 링크: https://leetcode.com/problems/evaluate-division

Author: skysign

leetcode.com 399. Evaluate Division/main.py

@@ -0,0 +1,59 @@
+from typing import List
+
+
+class Solution:
+    def calcEquation(self, equations: List[List[str]], values: List[float], queries: List[List[str]]) -> List[float]:
+        dicts = {}
+
+        for i in range(len(equations)):
+            d1, d2 = equations[i]
+            v = values[i]
+
+            if d1 not in dicts.keys():
+                dicts[d1] = {}
+
+            dicts[d1][d2] = v
+            dicts[d1][d1] = 1
+
+            if d2 not in dicts.keys():
+                dicts[d2] = {}
+
+            dicts[d2][d1] = 1 / v
+            dicts[d2][d2] = 1
+
+        answer = []
+
+        for q1, q2 in queries:
+            rtn = self.direct(dicts, q1, q2)
+
+            if None != rtn:
+                answer.append(rtn)
+                continue
+
+            rtn = self.dfs(dicts, q1, q2, 1, [])
+            answer.append(rtn)
+
+        return answer
+
+    def direct(self, dicts, q1, q2):
+        if q1 in dicts.keys():
+            if q2 in dicts[q1].keys():
+                return dicts[q1][q2]
+
+        return None
+
+    def dfs(self, dicts, bgn, end, v, visited: List[str]):
+        if bgn not in dicts.keys():
+            return -1
+
+        for mid in dicts[bgn].keys():
+            if mid not in visited:
+                if mid == end:
+                    v *= dicts[bgn][mid]
+                    return v
+                else:
+                    rtn = self.dfs(dicts, mid, end, v * dicts[bgn][mid], visited + [bgn])
+                    if rtn != -1:
+                        return rtn
+
+        return -1

leetcode.com 399. Evaluate Division/test_main.py

@@ -0,0 +1,42 @@
+from unittest import TestCase
+from main import Solution
+
+
+class TestSolution(TestCase):
+    def test1_calc_equation(self):
+        sln = Solution()
+        self.assertEqual(
+            [6.00000, 0.50000, -1.00000, 1.00000, -1.00000],
+            sln.calcEquation(
+                [["a", "b"], ["b", "c"]], [2.0, 3.0], [["a", "c"], ["b", "a"], ["a", "e"], ["a", "a"], ["x", "x"]]
+            )
+        )
+
+    def test2_calc_equation(self):
+        sln = Solution()
+        self.assertEqual(
+            [3.75000, 0.40000, 5.00000, 0.20000],
+            sln.calcEquation(
+                equations=[["a", "b"], ["b", "c"], ["bc", "cd"]], values=[1.5, 2.5, 5.0], queries=[["a", "c"], ["c", "b"], ["bc", "cd"], ["cd", "bc"]]
+            )
+        )
+
+    def test3_calc_equation(self):
+        sln = Solution()
+        self.assertEqual(
+            [0.50000, 2.00000, -1.00000, -1.00000],
+            sln.calcEquation(
+                equations=[["a", "b"]], values=[0.5], queries=[["a", "b"], ["b", "a"], ["a", "c"], ["x", "y"]]
+            )
+        )
+
+    def test18_calc_equation(self):
+        sln = Solution()
+        self.assertEqual(
+            [360.00000, 0.00833, 20.00000, 1.00000, -1.00000, -1.00000],
+            sln.calcEquation(
+                equations=[["x1", "x2"], ["x2", "x3"], ["x3", "x4"], ["x4", "x5"]],
+                values=[3.0, 4.0, 5.0, 6.0],
+                queries=[["x1", "x5"], ["x5", "x2"], ["x2", "x4"], ["x2", "x2"], ["x2", "x9"], ["x9", "x9"]]
+            )
+        )