面试热题(数组中的第K个最大元素)

给定整数数组 nums 和整数 k,请返回数组中第 k 个最大的元素。

请注意,你需要找的是数组排序后的第 k 个最大的元素,而不是第 k 个不同的元素。

复制代码
输入: [3,2,1,5,6,4] 和 k = 2
输出: 5

提到数组中最大元素,我们往往想到就是先给数组进行排序,然后取最大值,现在我们按照这个思路写一写代码

  • 首先判断入参是否合法
java 复制代码
f (nums == null || nums.length == 0) {
            return 0;
        }
  • 然后对数组进行排序
java 复制代码
Arrays.sort(nums);//默认排序方法时双基准快排,效率较高
  • 因为我们取的是第k个最大的元素

因为数组的长度是6,而k是2,我们所需要求的值的索引刚好是4,所以我们可以得出我们所需要推出的值是nums.length-k(在做题的过程中,如果需要确定关系式的这种情况,个人建议还是举出例子,然后亲自推导比较好一点)

java 复制代码
 return nums[nums.length  - k];

接下来,提到最大值,大家还能想到什么方法?是不是有种数据结构,特能自动的为我们进行数值的排序,不错,就是优先队列 ,我们可以先将数组中的元素都往优先队列中塞进去,然后poll k次就是我们所需要的值,我们直接上代码

java 复制代码
   public int findKthLargest(int[] nums, int k) {
       if(nums==null||nums.length==0){
           return 0;
       }
       //对比较器进行重写,从大到小,因为PriorityQueue的默认排序时升序排序
       PriorityQueue<Integer> queue=new PriorityQueue<>((a,b)->{
           return b -a;});
       for(int num:nums){
           queue.offer(num);
       }
       int res=0;
       while(k>0){
           res=queue.poll();
           k--;
       }
       return res;
    }
相关推荐
让我们一起加油好吗20 分钟前
【基础算法】01BFS
数据结构·c++·算法·bfs·01bfs
孤狼灬笑25 分钟前
机器学习十大经典算法解析与对比
人工智能·算法·机器学习
靠近彗星2 小时前
3.1 栈
数据结构·算法
sulikey2 小时前
一文彻底理解:如何判断单链表是否成环(含原理推导与环入口推算)
c++·算法·leetcode·链表·floyd·快慢指针·floyd判圈算法
Swift社区2 小时前
LeetCode 402 - 移掉 K 位数字
算法·leetcode·职场和发展
_码力全开_2 小时前
P1005 [NOIP 2007 提高组] 矩阵取数游戏
java·c语言·c++·python·算法·矩阵·go
墨染点香3 小时前
LeetCode 刷题【124. 二叉树中的最大路径和、125. 验证回文串】
算法·leetcode·职场和发展
Camel卡蒙3 小时前
红黑树详细介绍(五大规则、保持平衡操作、Java实现)
java·开发语言·算法