华为机试HJ58输入n个整数,输出其中最小的k个

华为机试HJ58输入n个整数,输出其中最小的k个

题目:

输出输入中最小的k个元素

想法:

通过快速排序对输入的元素进行从小到大的排序,输出前k个最小的元素

input_number = input().split(" ")

number_list = input().split(" ")

number_list = [int(n) for n in number_list]

def partition(arr, low, high):
    i = low - 1
    pivot = arr[high]
    for j in range(low, high):
        if arr[j] <= pivot:
            i += 1
            arr[i], arr[j] = arr[j], arr[i]
    arr[i+1], arr[high] = arr[high], arr[i+1]
    return i + 1


def quick_sort(arr, low, high):
    if low < high:
        pi = partition(arr, low, high)
        quick_sort(arr, low, pi-1)
        quick_sort(arr, pi+1, high)
    return arr
result = quick_sort(number_list, 0, len(number_list)-1)[:int(input_number[1])]
result = [str(r) for r in result]
print(" ".join(result))
  • 1
    点赞
  • 0
    收藏
    觉得还不错? 一键收藏
  • 0
    评论
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值