题解 | #最小的K个数#
最小的K个数
https://www.nowcoder.com/practice/6a296eb82cf844ca8539b57c23e6e9bf
#
# 代码中的类名、方法名、参数名已经指定，请勿修改，直接返回方法规定的值即可
#
#
# @param input int整型一维数组
# @param k int整型
# @return int整型一维数组
#
# 每次将保存的四个值中的最大值与当前值比较
# 若当前值更小，则更新答案
class Solution:
def GetLeastNumbers_Solution(self , input: List[int], k: int) -> List[int]:
if k == 0:
return None
elif k == 1:
return [min(input)]
ans = [1000 for _ in range(k)]
for i in input:
t = max(ans)
if i <= t:
ans[ans.index(t)] = i
return ans
投递字节跳动等公司8个岗位 >
腾讯云智研发稳定性 12人发布