一、经典的TOPK:最小的k个数
1. 采用快排的思想,先将数组进行排序,然后切片 取前k个
class Solution:
def getLeastNumbers(self, arr: List[int], k: int) -> List[int]:
def quicksort(arr,left,right) -> List[int]:
if left>right:
return
l,r = left,right
while l < r :
while l<r and arr[r]>=arr[left]:r-=1
while l<r and arr[l]<=arr[left]:l+=1
arr[l],arr[r] = arr[r],arr[l]
arr[left],arr[l] = arr[l],arr[left]
quicksort(arr,left,l-1)
quicksort(arr,r+1,right)
return arr
quicksort(arr,0,len(arr)-1)
return arr[:k]
2. 若题目中并没有要求 按顺序输出 前 k小的数据,可只需要将 数组划分为 最小的k个数 和 其余数据 两部分。
所以 在进行完一轮快排以后,就判断基准数 的坐标是否 l==k:
如果 l < k :说明左侧的数还不够k个 需要对右侧的数组 再次进行快排;
如果 l > k :说明左侧的数超过k个 需要对左侧的数组 再次进行快排;
class Solution:
def getLeastNumbers(self, arr: List[int], k: int) -> List[int]:
def quicksort(arr,left,right) -> List[int]:
if left>right:
return
l,r = left,right
while l < r :
while l<r and arr[r]>=arr[left]:r-=1
while l<r and arr[l]<=arr[left]:l+=1
arr[l],arr[r] = arr[r],arr[l]
arr[left],arr[l] = arr[l],arr[left]
if l==k:
return arr[:l]
elif l<k:
quicksort(arr,r+1,right)
else:
quicksort(arr,left,l-1)
quicksort(arr,0,len(arr)-1)
return arr[:k]