Update maximize-sum-of-array-after-k-negations.py

Author: kamyu104

Python/maximize-sum-of-array-after-k-negations.py

@@ -28,11 +28,11 @@ def PartitionAroundPivot(left, right, pivot_idx, nums, compare):
             while left <= right:
                 pivot_idx = random.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(A, K, lambda a, b: a < b)