python小课——零基础入门——学习笔记刷题学编程Leetcode模拟面试

LeetCode 面试题 17.14. 最小K个数

2021-09-22  本文已影响0人  freesan44

题目

设计一个算法,找出数组中最小的k个数。以任意顺序返回这k个数均可。

示例:

输入: arr = [1,3,5,7,2,4,6,8], k = 4
输出: [1,2,3,4]
提示:

0 <= len(arr) <= 100000
0 <= k <= min(100000, len(arr))

解题思路

class Solution:
    def smallestK(self, arr: List[int], k: int) -> List[int]:
        arr.sort()
        return arr[:k]

if __name__ == '__main__':
    arr = [1,3,5,7,2,4,6,8]
    k = 4
    ret = Solution().smallestK(arr, k)
    print(ret)
上一篇 下一篇

猜你喜欢

热点阅读