Update k-closest-points-to-origin.py

Author: kamyu104

Python/k-closest-points-to-origin.py

@@ -31,11 +31,11 @@ def PartitionAroundPivot(left, right, pivot_idx, nums, compare):
             while left <= right:
                 pivot_idx = randint(left, right)
                 new_pivot_idx = PartitionAroundPivot(left, right, pivot_idx, nums, compare)
-                if new_pivot_idx == k - 1:
+                if new_pivot_idx == k:
                     return
-                elif new_pivot_idx > k - 1:
+                elif new_pivot_idx > k:
                     right = new_pivot_idx - 1
-                else:  # new_pivot_idx < k - 1.
+                else:  # new_pivot_idx < k.
                     left = new_pivot_idx + 1
 
         kthElement(points, K, lambda a, b: dist(a) < dist(b))