python剑指offer最小的K个数
题目描述:
输入n个整数,找出其中最小的K个数。例如输入4,5,1,6,2,7,3,8这8个数字,则最小的4个数字是1,2,3,4,。
思路:
使用快排中的partition思想。
①我们设定partition函数的哨兵为key=lists[left],在partition函数中完成一轮比较的结果是,比key大的数都在其右边,比key小的数放在其左边。完成该轮后返回其left=right时left的值。
②我们判断left的值是比k大还是小:
如果left的值比k大,说明上轮partition之后,lists中前left个小的数在左边,其余的数在其右边,我们还需要把寻找范围缩小,下次找的时候只在数组前面left个数中找了。
如果left的值比k小,说明上轮partition之后,前left个数找的太少了,我们需要再往数组的后面找。
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 | # -*- coding: utf-8 -*- """ Content: 找最小的k个数 """ """ 思路: 包括2个部分function_partion和function """ def function_partion(lists, left, right): # 划分函数处理部分 key = lists[left] while left < right: while left < right and lists[right] > = key: right - = 1 lists[left] = lists[right] while left < right and lists[left] < = key: left + = 1 lists[right] = lists[left] lists[right] = key return left def function(lists, k): # 划分法主要函数部分 length = len (lists) left = 0 right = length - 1 index = function_partion(lists, left, right) while k ! = index: if index > k - 1 : right = index - 1 else : left = index + 1 index = function_partion(lists, left, right) return lists[ 0 :k] lists = [ 1 , 1 , 6 , 4 , 11 , 9 , 2 , 10 , 3 ] print ( "思路(划分法):" , function(lists, 4 )) |
【推荐】编程新体验,更懂你的AI,立即体验豆包MarsCode编程助手
【推荐】凌霞软件回馈社区,博客园 & 1Panel & Halo 联合会员上线
【推荐】抖音旗下AI助手豆包,你的智能百科全书,全免费不限次数
【推荐】博客园社区专享云产品让利特惠,阿里云新客6.5折上折
【推荐】轻量又高性能的 SSH 工具 IShell:AI 加持,快人一步