最小的k个数

最新推荐文章于 2021-05-28 15:34:40 发布

aaronthon

最新推荐文章于 2021-05-28 15:34:40 发布

阅读量115

点赞数

CC 4.0 BY-SA版权

分类专栏：数据结构与算法文章标签：算法 python leetcode 数据结构 java

本文链接：https://blog.youkuaiyun.com/aaronthon/article/details/109089542

数据结构与算法专栏收录该内容

45 篇文章

订阅专栏

本文介绍如何使用Python高效地解决寻找数组中最小K个数的问题，通过内置排序和选择算法，演示了如何在O(n log k)时间内找到答案。代码示例展示了如何在Solution类中实现GetLeastNumbers_Solution方法，适用于处理大规模数据。

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >

题目描述

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

解答

先排序，再返回最小的k个数

# coding:utf-8
class Solution:
    def GetLeastNumbers_Solution(self, tinput, k):
        # write code here
        for i in range(len(tinput)-1, 0, -1):
            for j in range(i):
                if tinput[j] > tinput[j+1]:
                    tinput[j], tinput[j+1] = tinput[j+1], tinput[j]
        if 0 < k <= len(tinput):
            return tinput[0:k]
        else:
            return []


p = Solution()
ret = p.GetLeastNumbers_Solution([2,1,3],3)
print ret

结束！