python 最小的k個數

2021-09-25 09:02:13 字數 1382 閱讀 3812

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 採用折半查詢 替換...