LeetCode.215 Kth Largest Element in an Array 返回数组中第K大的数

Question:

Find the kth largest element in an unsorted array. Note that it is the kth largest element in the sorted order, not the kth distinct element.

For example,
Given [3,2,1,5,6,4] and k = 2, return 5.

Note: 

You may assume k is always valid, 1 ≤ k ≤ array's length.

原题链接:https://leetcode.com/problems/kth-largest-element-in-an-array/description/

Java优先队列:

参考:https://blog.csdn.net/x_i_y_u_e/article/details/46381481

构造函数和API:

构造方法摘要

PriorityQueue() 
          使用默认的初始容量(11)创建一个 PriorityQueue,并根据其自然顺序对元素进行排序。

PriorityQueue(Collection<? extends E> c) 
          创建包含指定 collection 中元素的 PriorityQueue。

PriorityQueue(int initialCapacity) 
          使用指定的初始容量创建一个 PriorityQueue,并根据其自然顺序对元素进行排序。

PriorityQueue(int initialCapacity, Comparator<? super E> comparator) 
          使用指定的初始容量创建一个 PriorityQueue,并根据指定的比较器对元素进行排序。

PriorityQueue(PriorityQueue<? extends E> c) 
          创建包含指定优先级队列元素的 PriorityQueue。

PriorityQueue(SortedSet<? extends E> c) 
          创建包含指定有序 set 元素的 PriorityQueue。


方法摘要

 boolean

add(E e) 
          将指定的元素插入此优先级队列。

 void

clear() 
          从此优先级队列中移除所有元素。

 Comparator<? super E>

comparator() 
          返回用来对此队列中的元素进行排序的比较器;如果此队列根据其元素的自然顺序进行排序,则返回 null。

 boolean

contains(Object o) 
          如果此队列包含指定的元素,则返回 true。

 Iterator<E>

iterator() 
          返回在此队列中的元素上进行迭代的迭代器。

 boolean

offer(E e) 
          将指定的元素插入此优先级队列。

 E

peek() 
          获取但不移除此队列的头;如果此队列为空,则返回 null。

 E

poll() 
          获取并移除此队列的头,如果此队列为空,则返回 null。

 boolean

remove(Object o) 
          从此队列中移除指定元素的单个实例(如果存在)。

 int

size() 
          返回此 collection 中的元素数。

 Object[]

toArray() 
          返回一个包含此队列所有元素的数组。

<T> T[]

toArray(T[] a) 
          返回一个包含此队列所有元素的数组;返回数组的运行时类型是指定数组的类型。


思路:

可以使用优先队列的数据结构,维护一个有K个数的最大堆。遍历完数组之后,就可返回堆的头部元素,即使第K大的数。

参考:https://segmentfault.com/a/1190000003704825

class Solution {
    public int findKthLargest(int[] nums, int k) {
        PriorityQueue<Integer> que = new PriorityQueue<Integer> ();
        for (int i = 0; i < nums.length; i++)
        {
            que.add(nums[i]);
            if(que.size() > k) 
                que.poll();
        }
        return que.poll();
    }
}
博主学习记录,转载请注明出处,有错望指正,谢谢~^-^


### 关于LeetCode215题的描述 LeetCode215题名为 **Kth Largest Element in an Array**,其问题是要求在一个无序数组中找到第k的元素。需要注意的是,这里的“第k”是指按照降序排列后的第k个位置上的。 #### 解决方案概述 一种常见的解决方法是利用堆排序算法来实现这一目标。通过构建一个小顶堆(Min Heap),可以有效地获取到所需的第k元素。这种方法的时间复杂度通常为 \(O(n \log k)\),其中 n 是数组长度,而 k 则是我们要找的目标次序[^4]。 以下是基于 Python 的解决方案代码: ```python import heapq def findKthLargest(nums, k): # 使用heapq模块中的nlargest直接找出前k返回最后一个即为我们想要的结果 return heapq.nlargest(k, nums)[-1] # 测试样例 nums = [3, 2, 1, 5, 6, 4] k = 2 print(findKthLargest(nums, k)) # 输出应为5 ``` 此段代码借助了 `heapq` 库里的 `nlargest` 方法简化操作流程,从而达到快速定位的目的[^5]。 另外还有一种方式就是先对整个列表完成全面排序之后再选取相应索引处值作为最终答案;不过这种做法虽然直观却可能带来不必要的计算负担,在性能上未必优于上述提到过的最小堆策略。 ### 提供更高效的解答思路 除了运用内置库外还可以手动创建最堆(MaxHeap), 并持续移除顶部直到剩下最后那个代表所需值为止; 或者采用分治法(Divide And Conquer Approach)像快速选择(Quickselect Algorithm)那样只关注局部区域进而减少整体迭代次以提高效率至平均情况下的线性时间复杂度 O(N)[^6].
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值