自拍偷在线精品自拍偷,亚洲欧美中文日韩v在线观看不卡

詳解堆排序解決TopK問題

開發(fā) 前端
上次介紹了堆排序,這次介紹堆排序常見的應(yīng)用場景TopK問題。

[[377078]]

 上次介紹了堆排序,這次介紹堆排序常見的應(yīng)用場景TopK問題。

利用堆求TopK問題TopK問題是一個堆排序典型的應(yīng)用場景。

題目是這樣的:假設(shè),我們想在大量的數(shù)據(jù),如 100 億個整型數(shù)據(jù)中,找到值最大的 K 個元素,K 小于 10000。對此,你會怎么做呢?

對標(biāo)的是Leetcode第215題:「數(shù)組中的第K個最大元素?!?/strong>

具體鏈接:https://leetcode-cn.com/problems/kth-largest-element-in-an-array/

在未排序的數(shù)組中找到第 k 個最大的元素。請注意,你需要找的是數(shù)組排序后的第 k 個最大的元素,而不是第 k 個不同的元素。

  1. 示例 1: 
  2.  
  3. 輸入: [3,2,1,5,6,4] 和 k = 2 
  4. 輸出: 5 
  5. 示例 2: 
  6.  
  7. 輸入: [3,2,3,1,2,4,5,5,6] 和 k = 4 
  8. 輸出: 4 

經(jīng)典的TopK問題還有:最大(小) K 個數(shù)、前 K 個高頻元素、第 K 個最大(小)元素

對此TopK問題本質(zhì)上是一個排序問題,排序算法一共有十個,這個還有很多排序算法沒有介紹過。


至于為什么TopK問題最佳的答案是堆排序?其實在空間和時間的復(fù)雜度來考量,雖說快排是最好的排序算法,但是對于100億個元素從大到小排序,然后輸出前 K 個元素值。

可是,無論我們掌握的是快速排序算法還是堆排序算法,在排序的時候,都需要將全部的元素讀入到內(nèi)存中。也就是說,100億個整型元素大約需要占用40GB的內(nèi)存空間,這聽起來就不像是普通民用電腦能干的事情,(一般的民用電腦內(nèi)存比這個小,比如我寫文章用的電腦內(nèi)存是 32GB)。

眾所周知,快速排序和堆排序的時間復(fù)雜度都可以達到,但是對于快速排序來說,數(shù)據(jù)是順序訪問的。而對于堆排序來說,數(shù)據(jù)是跳著訪問的。比如堆排序中,最重要的一個操作就是數(shù)據(jù)的堆化。因此,快速排序的時間復(fù)雜度是優(yōu)于堆排序的。

但是快速排序是新建數(shù)組,空間復(fù)雜度是,遠低于堆排序的。對于龐大的數(shù)據(jù)量,應(yīng)該優(yōu)先選擇堆排序。

如果使用heapq內(nèi)置模塊,尋找數(shù)組中的第K個最大元素就是一行代碼,heapq中的nlargest接口封裝好了,返回的是一個數(shù)組,需要切片取值。

  1. import heapq 
  2. class Solution: 
  3.     def findKthLargest(self, nums: List[int], k: int) -> int
  4.         return heapq.nlargest(k,nums)[-1] 

當(dāng)然,一般都是手寫堆排序,尋找數(shù)組中的第K個最大元素建立最小堆,尋找數(shù)組中的第K個最小元素建立最大堆,

思路:「取nums前K個元素建立大小為K的最小堆,后面就是維護一個容量為k的小頂堆,堆中的k個節(jié)點代表著當(dāng)前最大的k個元素,而堆頂顯然是這k個元素中的最小值?!?/strong>

因此只要遍歷整個數(shù)組,當(dāng)二叉堆大小等于K后,當(dāng)遇見大于堆頂數(shù)值的元素時彈出堆頂,并壓入該元素,持續(xù)維護最大的K個元素。遍歷結(jié)束后,堆頂元素即為第K個最大元素。時間復(fù)雜度。

  1. class Solution: 
  2.     def findKthLargest(self, nums: List[int], k: int) -> int
  3.         heapsize=len(nums) 
  4.         def maxheap(a,i,length): 
  5.             l=2*i+1 
  6.             r=2*i+2 
  7.             large=i 
  8.             if l<length and a[l]>a[large]: 
  9.                 large=l 
  10.             if r<length and a[r]>a[large]: 
  11.                 large=r 
  12.             if large!=i: 
  13.                 a[large],a[i]=a[i],a[large] 
  14.                 maxheap(a,large,length) 
  15.              
  16.         def buildheap(a,length): 
  17.             for i in range(heapsize//2,-1,-1): 
  18.                 maxheap(a,i,length) 
  19.  
  20.         buildheap(nums,heapsize) 
  21.         for i in range(heapsize-1,heapsize-k,-1): 
  22.             nums[0],nums[i]=nums[i],nums[0] 
  23.             heapsize-=1 
  24.             maxheap(nums,0,heapsize) 
  25.         return nums[0]   

相反如果是求前k個最小,那么就用最大堆,因此面對TopK問題,最完美的解法是堆排序。因此,只有你看到數(shù)組的第K個……,馬上就是想到堆排序。

如果在數(shù)據(jù)規(guī)模小、對時間復(fù)雜度、空間復(fù)雜度要求不高的時候,真沒必要上 “高大上” 的算法,寫一個快排就很完美了。

TopK問題就像搜索引擎每天會接收大量的用戶搜索請求,它會把這些用戶輸入的搜索關(guān)鍵詞記錄下來,然后再離線地統(tǒng)計分析,得到最熱門的Top10搜索關(guān)鍵詞,啥啥惹事就出來了。

本文已收錄 GitHub https://github.com/MaoliRUNsen/runsenlearnpy100

 

責(zé)任編輯:姜華 來源: Python之王
相關(guān)推薦

2011-04-20 15:06:44

堆排序

2021-01-19 07:02:26

算法數(shù)據(jù)結(jié)構(gòu)堆排序

2014-10-30 15:59:10

2020-03-06 16:08:46

堆結(jié)構(gòu)堆排序應(yīng)用

2021-12-20 10:39:30

TopK排序代碼

2021-03-23 08:33:22

Java數(shù)據(jù)結(jié)構(gòu)算法

2023-10-10 08:00:07

2021-12-14 10:54:31

TopK面試排序法

2009-03-04 13:32:28

排序SQLOracle

2009-11-17 09:17:49

PHP多維數(shù)組排序

2009-11-09 15:10:42

2011-08-16 13:34:23

Objective-C歸檔

2009-09-10 16:30:11

C#排序函數(shù)

2022-02-18 08:26:12

TopK數(shù)組面試題

2023-10-05 06:02:52

計數(shù)排序Counting

2010-08-03 09:12:52

Flex安全沙箱

2017-06-14 22:11:57

數(shù)據(jù)庫MySQL死鎖

2010-11-11 14:05:17

SQL Server排

2021-01-04 08:39:26

JAVA8OptionalNPE

2009-02-18 14:28:23

編碼亂碼JSP
點贊
收藏

51CTO技術(shù)棧公眾號