-
Notifications
You must be signed in to change notification settings - Fork 0
/
Copy path215_1.py
36 lines (28 loc) · 834 Bytes
/
215_1.py
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
class Solution:
def findKthLargest(self, nums: List[int], k: int) -> int:
if k > len(nums):
return
t = k - 1
start = 0
end = len(nums) - 1
while True:
r = self.sort_one(nums, start, end)
if r == t:
return nums[t]
elif r < t:
start = r + 1
else:
end = r - 1
return nums[t]
def sort_one(self, nums, start, end):
if start >= end:
return start
s, e = start, end
while s < e:
while s < e and nums[s] >= nums[e]:
e -= 1
nums[s], nums[e] = nums[e], nums[s]
while s < e and nums[s] >= nums[e]:
s += 1
nums[s], nums[e] = nums[e], nums[s]
return s