对于一个无序数组,数组中元素为互不相同的整数,请返回其中最小的k个数,顺序与原数组中元素顺序一致。
给定一个整数数组A及它的大小n,同时给定k,请返回其中最小的k个数。
[1,2,4,3],4,2
返回:[1,2]
# -*- coding:utf-8 -*- class KthNumbers: def findKthNumbers(self, A, n, k): B=sorted(A)#对A进行排序 ans=filter(lambda x:x<=B[k-1],A)#过滤出小于等于第k大的数字 return ans
我大python代码只要写在return里面,只需要一行就***:
return sorted(sorted(A)[:k], key=A.index)
这道题你会答吗?花几分钟告诉大家答案吧!
扫描二维码,关注牛客网
下载牛客APP,随时随地刷题