天天看点

一文详解面试常考的TopK问题

首发公众号:<code>bigsai</code> ,转载请附上本文链接

hello,大家好,我是bigsai哥哥,好久不见,甚是想念哇🤩!

今天给大家分享一个TOPK问题,不过我这里不考虑特别大分布式的解决方案,普通的一道算法题。

首先搞清楚,什么是topK问题?

topK问题,就是找出序列中前k大(或小)的数,topK问题和第K大(或小)的解题思路其实大致一致的。

TopK问题是一个非常经典的问题,在笔试和面试中出现的频率都非常非常高(从不说假话)。下面,从小小白的出发点,认为topK是求前K大的问题,一起认识下TopK吧!

当前,在求TopK和第K大问题解法差不多,这里就用力扣215数组的第k个大元素 作为解答的题演示啦。学习topk之前,这篇程序员必知必会的十大排序一定要会。

找到TopK,并且排序TopK

啥,你想要我找到TopK?不光光TopK,你想要多少个,我给你多少个,并且还给你排序给排好,啥排序我最熟悉呢?

如果你想到冒泡排序O(n^2)那你就大意了啊。

如果使用O(n^2)级别的排序算法,那也是要优化的,其中冒泡排序和简单选择排序,每一趟都能顺序确定一个最大(最小)的值,所以不需要把所有的数据都排序出来,只需要执行K次就行啦,所以这种算法的时间复杂度也是O(nk)。

这里给大家回顾一下冒泡排序和简单选择排序区别:

冒泡排序和简单选择排序都是多趟,每趟都能确定一个最大或者最小,区别就是冒泡在枚举过程中只和自己后面比较,如果比后面大那么就交换;而简单选择是每次标记一个最大或者最小的数和位置,然后用这一趟的最后一个位置数和它交换(每一趟确定一个数枚举范围都慢慢变小)。

下面用一张图表示过程:

一文详解面试常考的TopK问题

这里把code也给大家提供一下,简单选择上面图给的是每次选最小,实现的时候每次选最大就可以了。

当然,快排和归并排序甚至堆排序也可以啊,这些排序的时间复杂度为O(nlogn),也就是将所有数据排序完然后直接返回结果,这部分就不再详细讲解啦,调调api或者手写排序都可。

两种思路的话除了K极小的情况O(nk)快一些,大部分情况其实还是O(nlogn)情况快一些的,不过从O(n^2)想到O(nk),还是有所收获的。

这里需要知道堆相关的知识,我以前写过优先队列和堆排序,这里先不重复讲,大家也可以看一下:

优先队列不知道,看看堆排序吧

硬核,手写一个优先队列

上面说道堆排序O(nlogn)那是将所有元素都排序完然后取前k个,但是其实上我们分析一下这个堆排序的过程和几个注意点哈:

堆这种数据结构,分为大根堆和小根堆,小根堆是父节点值小于子节点值,大根堆是父节点的值大于子节点的值,这里肯定是要采用大根堆的。

堆看起来是一个树形结构,但是堆是个完全二叉树我们用数组存储效率非常高,并且也非常容易利用下标直接找到父子节点,所以都用数组来实现堆,每次排序完成的节点都将数移到数组末尾让一个新数组组成一个新的堆继续。

堆排序从大的来看可以分成两个部分,无序数组建堆和在堆基础上每次取对顶排序。其中无序数组建堆的时间复杂度为O(n),在堆基础上排序每次取堆顶元素,然后将最后一个元素移到堆顶进行调整堆,每次只需要O(logn)级别的时间复杂度,完整排序完n次就是O(nlogn),但是咱们每次只需要k次,所以完成k个元素排序功能需要花费O(klogn)时间复杂度,整个时间复杂度为O(n+klogn)因为和前面区分一下就不合并了。

画了一张图帮助大家理解,进行两次就获得Top2,进行k次就获得TopK了。

一文详解面试常考的TopK问题

实现代码为:

排序总有一些骚操作的排序—线性排序,那么你可能会问桶类排序可以嘛?

也可以啦,不过要看数值范围进行优化,桶类排序适合数据均匀密集出现次数比较多的情况,而计数排序更是希望数值能够小一点。

那么利用桶类排序的具体核心思想是怎么样的呢?

先用计数排序统计各个数字出现次数,然后将新开一个数组从后往前叠加求和计算。

一文详解面试常考的TopK问题

这种情况非常适合数值巨量并且分布范围不大的情况。

代码本来不想写了,但是念在你会给我三连我写一下吧

好啦,今天的TopK问题就到这里啦,相信你下次遇到肯定会拿捏它。

TopK问题不难,就是巧妙利用排序而已。排序是非常重要的,面试会非常高频。

这里我就不藏着掖着摊牌了,以面试官的角度会怎么引导你说TOPK问题。

狡猾的面试官:

嗯,我们来聊聊数据结构与算法,来讲讲排序吧,你应该接触过吧?讲出你最熟悉的三种排序方式,并讲解一下其中具体算法方式。

卑微的我:

bia la bia la bia la bia la……

如果你提到快排,桶排序说不定就让你用这个排序实现一下TopK问题,其他排序也可能,所以掌握好十大排序是非常必要的!

个人原创公众号:<code>bigsai</code> 欢迎关注,花了半年写了一本原创数据结构与算法pdf。

一文详解面试常考的TopK问题