Python 最小的 k 個數

題目描述

輸入n個整數，找出其中最小的k個數。例如輸入4,5,1,6,2,7,3,8這8個數字，則最小的4個數字是1,2,3,4,。

# -*- coding:utf-8 -*-

'''最小的k個數
題目描述
輸入n個整數，找出其中最小的k個數。例如輸入4,5,1,6,2,7,3,8這8個數字，則最小的4個數字是1,2,3,4,。
'''class solution:
# 蒂姆排序：
def getleastnumbers_solution1(self, tinput, k):
# write code here
if tinput ==  or k > len(tinput):
return 
tinput.sort()
return tinput[: k]
# 快速排序：
def getleastnumbers_solution2(self, tinput, k):
def quick_sort(lst):
if not lst:
return 
pivot = lst[0]
left = quick_sort([x for x in lst[1: ] if x < pivot])
right = quick_sort([x for x in lst[1: ] if x >= pivot])
return left + [pivot] + right
if tinput ==  or k > len(tinput):
return 
tinput = quick_sort(tinput)
return tinput[: k]
# 歸併排序：
def getleastnumbers_solution3(self, tinput, k):
def merge_sort(lst):
if len(lst) <= 1:
return lst
mid = len(lst) // 2
left = merge_sort(lst[: mid])
right = merge_sort(lst[mid:])
return merge(left, right)
def merge(left, right):
l, r, res = 0, 0, 
while l < len(left) and r < len(right):
if left[l] <= right[r]:
l += 1
else:
r += 1
res += left[l:]
res += right[r:]
return res
if tinput ==  or k > len(tinput):
return 
tinput = merge_sort(tinput)
return tinput[: k]
# 堆排序：
def getleastnumbers_solution4(self, tinput, k):
def siftup(lst, temp, begin, end):
if lst == :
return 
i, j = begin, begin * 2 + 1
while j < end:
if j + 1 < end and lst[j + 1] > lst[j]:
j += 1
elif temp > lst[j]:
break
else:
lst[i] = lst[j]
i, j = j, 2 * j + 1
lst[i] = temp
def heap_sort(lst):
if lst == :
return 
end = len(lst)
for i in range((end // 2) - 1, -1, -1):
siftup(lst, lst[i], i, end)
for i in range(end - 1, 0, -1):
temp = lst[i]
lst[i] = lst[0]
siftup(lst, temp, 0, i)
return lst
if tinput ==  or k > len(tinput):
return 
tinput = heap_sort(tinput)
return tinput[: k]
# 氣泡排序：
def getleastnumbers_solution5(self, tinput, k):
def bubble_sort(lst):
if lst == :
return 
for i in range(len(lst)):
for j in range(1, len(lst) - i):
if lst[j-1] > lst[j]:
lst[j-1], lst[j] = lst[j], lst[j-1]
return lst
if tinput ==  or k > len(tinput):
return 
tinput = bubble_sort(tinput)
return tinput[: k]
# 直接選擇排序：
def getleastnumbers_solution6(self, tinput, k):
def select_sort(lst):
if lst == :
return 
for i in range(len(lst)-1):
smallest = i
for j in range(i, len(lst)):
if lst[j] < lst[smallest]:
smallest = j
lst[i], lst[smallest] = lst[smallest], lst[i]
return lst
if tinput ==  or k > len(tinput):
return 
tinput = select_sort(tinput)
return tinput[: k]
# 插入排序：
def getleastnumbers_solution7(self, tinput, k):
def insert_sort(lst):
if lst == :
return 
for i in range(1, len(lst)):
temp = lst[i]
j = i
while j > 0 and temp < lst[j - 1]:
lst[j] = lst[j - 1]
j -= 1
lst[j] = temp
return lst
if tinput ==  or k > len(tinput):
return 
tinput = insert_sort(tinput)
return tinput[: k]

最小的K個數（Python）

題目描述 輸入n個整數，找出其中最小的k個數。例如輸入4,5,1,6,2,7,3,8這8個數字，則最小的4個數字是1,2,3,4,這個題目完成的思路有很多，很多排序演算法都可以完成既定操作，關鍵是複雜度性的考慮。以下幾種思路當是筆者拋磚引玉，如果讀者有興趣可以自己再使用其他方法一一嘗試。思路1 利用...

python 最小的k個數

class solution def getleastnumbers solution self,t,k write code here if len t 0 or len t k or k 0 不設定k 0 返回，k為0時會陷入無限迴圈 return s 0 l len t 1 index sel...

Python 最小的K個數

輸入n個整數，找出其中最小的k個數。例如輸入4,5,1,6,2,7,3,8這8個數字，則最小的4個數字是1,2,3,4 最大堆的查詢時間複雜度為o logk 替換的複雜度也為o logk 輔助陣列的空間複雜度為o k 如果換為用陣列解決該問題，那麼 查詢的時間複雜度為o logk 採用折半查詢 替換...