Update kth_largest_element.py

data_structures/arrays/kth_largest_element.py

@@ -99,19 +99,18 @@
         raise ValueError("Invalid value of 'position'")
     low, high = 0, len(arr) - 1
     while low <= high:
-        if low > len(arr) - 1 or high < 0:
-            return -1
         pivot_index = partition(arr, low, high)
-        if pivot_index == position - 1:
+         target_index = position - 1
+        if pivot_index == target_index:
             return arr[pivot_index]
-        elif pivot_index > position - 1:
+        elif pivot_index > target_index:
             high = pivot_index - 1
         else:
             low = pivot_index + 1
     return -1
 
 
 if __name__ == "__main__":
     import doctest
 
     doctest.testmod()