堆排序的过程:
1.创建初始堆。
2.将堆的最后一个元素和堆顶元素交换,减少堆的范围, 调整堆。
class Solution:
def findKthLargest(self, nums: List[int], k: int) -> int:
len_nums = len(nums)
for i in range(len_nums//2-1, -1, -1):
self.adjust_heap(nums, i, len_nums) # 从最后一个非叶子节点向上调整堆
for j in range(len_nums-1, 0, -1):
nums[0], nums[j] = nums[j], nums[0] # 交换堆顶元素和堆最后一个元素
self.adjust_heap(nums, 0, j) # 缩小堆的范围,调整堆顶
return nums
def adjust_heap(self, target_list, i, target_len):
k = 2 * i + 1 # 左孩子坐标
tmp = target_list[i]
while k < target_len:
if k+1 < target_len and target_list[k+1] > target_list[k]:
k += 1
if tmp < target_list[k]: # 需要和tmp比,因为i在变化
target_list[i] = target_list[k]
i = k
else: # 根据堆的性质,后面的不需要继续比了
break
k = 2 * k + 1
target_list[i] = tmp