题目描述
输入n个整数,找出其中最小的K个数。例如输入4,5,1,6,2,7,3,8这8个数字,则最小的4个数字是1,2,3,4,。
自己的思路就是如果k<n时,对n个数进行排序就可以知道满足条件的数字是哪些了;如果k>n,则返回False
# -*- coding:utf-8 -*-
class Solution:
def GetLeastNumbers_Solution(self, tinput, k):
# write code here
s=[]
lenT=len(tinput)
if k>lenT:
return s
else:
s=sorted(tinput)
return s[0:k]
如果面试时,不让用排序该怎么解决呢?
1.利用快速排序
如果基于数组的第k个数字来调整,则使得比k个数字小的所有数字都位于数组的左边,比k个数字大的所有数字都位于数组的右边。这样调整之后,位于数组中左边的k个数字就是最小的k个数字(这k个数字不一定是排序的)这种方法会改变原始数组
2.利用容器
容器中先放入k个数,如果第k+1个数进来的时候要判断这个数是否比容器中所有的数都大,如果是这样则抛弃这个数,否则,将其替换掉容器中最大的数,依次类推,知道最后整个序列遍历完成。但在剑指offer中提到可以利用堆每次来选择容器中最大的数字但实现比较复杂,可以使用红黑树。这种方法可以用来处理海量数据中找到最小的k个数字。