面试高频算法难题:数组中的第K个最大元素
目是什么意思呢?比如给定的无序数组如下:
如果 k=6,也就是要寻找数组中的第6大元素,这个元素是哪一个呢?
显然,数组中第一大的元素是24,第二大的元素是20,第三大的元素是17 …第6大的元素是9。
让我想想啊……
对了,我可以先把无序数组排序,然后数出排序后的第k个元素!
方法1:排序法
这是最容易想到的方法,先把无序数组从大到小进行排序,排序后的第k个元素,自然就是数组中的第k大元素。
先进行排序的话,算法时间复杂度是O(nlogn),
性能有些差,有没有更优化的方法?
让我想想啊……
对了,我可以维护一个长度为k的数组,有序存储当前k个较大的元素!
方法2:插入法
维护一个长度为k的有序数组A,用于存储已知的k个较大的元素。
接下来遍历原数组,每遍历到一个元素,和数组A中最小的元素相比较,如果小于等于数组A的最小元素,继续遍历;如果大于数组A的最小元素,则插入到数组A中,并把曾经的最小元素“挤出去”。
比如k=3,先把最左侧的7,5,15三个数有序放入数组A当中,代表当前最大的3个数。
这时候,遍历到元素3, 由于3<5,继续遍历。
接下来遍历到17,由于17>5,插入到数组A的合适位置,类似于插入排序,并把原先最小的元素5“挤出去”。
继续遍历原数组,一直遍历到数组的最后一个元素…
最终,数组A中存储的元素是24,20,17,代表着整个数组中最大的3个元素。此