有关排序题目合集

一、经典的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]