leetcode.com 399. Evaluate Division

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

Author: skysign

leetcode.com 399. Evaluate Division/main.py

@@ -24,28 +24,16 @@ def calcEquation(self, equations: List[List[str]], values: List[float], queries:
         answer = []
 
         for q1, q2 in queries:
-            rtn = self.direct(dicts, q1, q2)
-
-            if None != rtn:
-                answer.append(rtn)
+            if q1 not in dicts.keys():
+                answer.append(-1)
                 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:

leetcode.com 399. Evaluate Division/test_main.py

@@ -33,7 +33,7 @@ def test3_calc_equation(self):
     def test18_calc_equation(self):
         sln = Solution()
         self.assertEqual(
-            [360.00000, 0.00833, 20.00000, 1.00000, -1.00000, -1.00000],
+            [360.00000, 0.008333333333333333, 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],