快排partition

最新推荐文章于 2023-04-10 20:38:13 发布

Massacre96Wj

最新推荐文章于 2023-04-10 20:38:13 发布

阅读量164

点赞数

分类专栏：算法文章标签：排序算法算法数据结构

本文链接：https://blog.youkuaiyun.com/weixin_42047706/article/details/124867779

版权

算法专栏收录该内容

6 篇文章

订阅专栏

本文介绍了两种快速排序的实现方式，partition函数分别从两端和起始位置遍历。还展示了如何利用快速排序的思想解决TopK问题，通过调整分区过程找到第K小的元素。这些算法在数据处理和排序场景中非常实用。

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >

快速排序partition函数

def partition(arr, left, right):
  '''
  左右遍历
  返回pivot的位置索引
  '''
  pivot = arr[left]
  while left < right:
    while left < right and arr[right] > pivot:
      right -= 1
    arr[left] = arr[right]
    while left < right and arr[left] <= pivot:
      left += 1
    arr[right] = arr[left]
  arr[left] = pivot
  return left

def partition2(arr, left, right):
  '''
  从开头遍历
  '''
  pivot = arr[left]
  j = left
  for i in range(left, right):
    if arr[i] < pivot:
      j += 1
      arr[j], arr[i] = arr[i], arr[j]
  arr[j], arr[left] = arr[left], arr[j]
  return j

应用一、快排序

def quicksort(arr, left, right):
  if left < right:
    position = partition(arr, left, right)
  quicksort(arr, left, position-1)
  quicksort(arr, position+1, right)

应用二、Topk

def findTopK(arr, k):
  left, right = 0, len(arr) - 1
  k = len(arr) - k
  while left < right:
    pos = partition(arr, left, right)
    if pos == k:
      return arr[k]
    elif pos < k:
      left = pos + 1
    else:
      right = pos - 1