輸入n個整數,找出其中最小的k個數。例如輸入4,5,1,6,2,7,3,8這8個數字,則最小的4個數字是1,2,3,4。
示例1
[4,5,1,6,2,7,3,8],4
[1,2,3,4]思路:使用堆排序實現
解法1:自己構建堆
解法2:使用python內建的sorted函式實現:sorted函式是不穩定的排序方法
# -*- coding:utf-8 -*-
class solution:
def getleastnumbers_solution(self, tinput, k):
# write code here
if(k>len(tinput)):return
return sorted(tinput)[:k]
解法3:使用python內建的堆heapq實現
# -*- coding:utf-8 -*-
import heapq
class solution:
def getleastnumbers_solution(self, tinput, k):
# write code here
if(k>len(tinput)):return
return heapq.nsmallest(k, tinput)
heapq堆的一些操作可見: 劍指offer 最小k個數
1.題目 輸入n個整數,找出其中最小的k個數。例如輸入4,5,1,6,2,7,3,8這8個數字,則最小的4個數字是1,2,3,4,2.方法 1 基於堆排序演算法,構建最大堆。時間複雜度為o nlogk 2 如果用快速排序,時間複雜度為o nlogn 3 如果用插入排序,時間複雜度為o n 2 3.演...
劍指offer 最小的K個數
華電北風吹 天津大學認知計算與應用重點實驗室 日期 2015 10 4 題目描述 輸入n個整數,找出其中最小的k個數。例如輸入4,5,1,6,2,7,3,8這8個數字,則最小的4個數字是1,2,3,4,解析 基於插入排序的思想可以想到使用乙個長度為k的排序陣列儲存最小的k個元素,複雜度o nk 基於...
《劍指offer》最小的K個數
輸入n個整數,找出其中最小的k個數。例如輸入4,5,1,6,2,7,3,8這8個數字,則最小的4個數字是1,2,3,4,輸入給整數陣列,和k 找出其中最小的k個數 class solution 1 先用插入排序對陣列中數進行排序。2 取出有序陣列中最小的kge。附 這只是乙個比較簡單的方法。如果考慮...