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 = self.partition(t, s, l)
print('index1', index)
while index != (k - 1) :
if index > k - 1:
l = index - 1
#必須寫為l = index - 1, s = index + 1
#否則s與partition返回的index相等時,會陷入無限迴圈
elif index < k - 1:
s = index + 1
index = self.partition(t, s, l)
print('index2',index)
return t[:k]
def partition(self, t, s, l):
p = t[s]
print('s,l',s,l)
print(t)
t[l], t[s] = t[s], t[l]
print(t)
small = s - 1
large = s
while large < l:
if t[large] < p:
small += 1
if large != small:
t[small], t[large] = t[large], t[small]
large += 1
# while t[large] >= p and large < l:
# large += 1
# if large != small:
# print(small, large)
# t[small], t[large] = t[large], t[small]
# small += 1
## print('hi:',t)
# large += 1
small += 1
t[small], t[l] = t[l], t[small]
#print(t)
print(small)
return small
注意以上陷入無限迴圈的點。
求解最小的k個數,兩種方法:1)先排序,在取前k個,快排最佳 ,以上便是 2)最小堆
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,...
最小的K個數(Python)
題目描述 輸入n個整數,找出其中最小的k個數。例如輸入4,5,1,6,2,7,3,8這8個數字,則最小的4個數字是1,2,3,4,這個題目完成的思路有很多,很多排序演算法都可以完成既定操作,關鍵是複雜度性的考慮。以下幾種思路當是筆者拋磚引玉,如果讀者有興趣可以自己再使用其他方法一一嘗試。思路1 利用...
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 採用折半查詢 替換...