可以使用快速選擇算法來實現求第k大的數。快速選擇算法的思想是基于快速排序,通過每次將數組劃分為兩個部分,將選定的元素放在正確的位置上,從而找到第k大的數。
下面是使用快速選擇算法實現求第k大的數的Python代碼:
def partition(nums, left, right):
pivot = nums[left]
i = left + 1
j = right
while True:
while i <= j and nums[i] >= pivot:
i += 1
while i <= j and nums[j] <= pivot:
j -= 1
if i <= j:
nums[i], nums[j] = nums[j], nums[i]
else:
break
nums[left], nums[j] = nums[j], nums[left]
return j
def findKthLargest(nums, k):
left = 0
right = len(nums) - 1
while True:
pos = partition(nums, left, right)
if pos == k - 1:
return nums[pos]
elif pos > k - 1:
right = pos - 1
else:
left = pos + 1
使用示例:
nums = [3, 2, 1, 5, 6, 4]
k = 2
result = findKthLargest(nums, k)
print(result) # 輸出: 5
在以上示例中,我們給定了一個數組nums
和一個整數k
,然后調用findKthLargest
函數來求第k大的數。輸出結果為5,表示數組nums
中第2大的數為5。