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(CollectionE> c) 
          创建包含指定 collection 中元素的 PriorityQueue。

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

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

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

PriorityQueue(SortedSetE> c) 
          创建包含指定有序 set 元素的 PriorityQueue。


方法摘要

 boolean

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

 void

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

 ComparatorE>

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[]

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


思路:

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

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

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


你可能感兴趣的:(Leetcode笔记)