最小的k个数

题目描述

输入n个整数,找出其中最小的K个数。例如输入4,5,1,6,2,7,3,8这8个数字,则最小的4个数字是1,2,3,4,。

自己的思路就是如果kn,则返回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个数字。

你可能感兴趣的:(牛客网刷题)