Update median-of-two-sorted-arrays.py

kamyu104 · kamyu104 · commit d7bbdabff77d · 2016-05-28T22:17:12.000+08:00
diff --git a/Python/median-of-two-sorted-arrays.py b/Python/median-of-two-sorted-arrays.py
@@ -65,13 +65,13 @@ def binary_search(array, left, right, target, compare):
                     left = mid + 1
             return left
 
-        def match(arrays, target, k):
+        def match(arrays, num, target):
             res = 0
             for array in arrays:
                 if array:
-                    res += len(array) - binary_search(array, 0, len(array) - 1, target, \
+                    res += len(array) - binary_search(array, 0, len(array) - 1, num, \
                                                       lambda array, x, y: array[x] > y)
-            return res < k
+            return res < target
 
         left, right = float("inf"), float("-inf")
         for array in arrays:
