最小的k个数
输入整数数组 arr ,找出其中最小的 k 个数。例如,输入4、5、1、6、2、7、3、8这8个数字,则最小的4个数字是1、2、3、4。
from typing import List, Tuple
import collections
from collections import deque
class Solution:
def getLeastNumbers(self, arr: List[int], k: int) -> List[int]:
def qiuck_sort(arr,l,r):
if l >= r:
return
i,j=l,r
while i<j:
while i<j and arr[j]>=arr[l]:
j -=1
while i<j and arr[i]<=arr[l]:
i +=1
arr[i],arr[j]=arr[j],arr[i]
arr[l],arr[i]=arr[i],arr[l]
qiuck_sort(arr,l,i-1)
qiuck_sort(arr,i+1,r)
qiuck_sort(arr,0,len(arr)-1)
print(arr)
return arr
a=[3,2,1,6,4]
b=Solution().getLeastNumbers(a,2)
print(b)