LeetCode - Kth Largest Element in an Array
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. Example 1: Input: [3,2,1,5,6,4] and k = 2 Output: 5 Example 2: Input: [3,2,3,1,2,4,5,5,6] and k = 4 Output: 4 Note: You may assume k is always valid, 1 ≤ k ≤ array's length.
用 PriorityQueue
class Solution { public int findKthLargest(int[] nums, int k) { if (nums == null || nums.length == 0) { return -1; } PriorityQueue<Integer> pq = new PriorityQueue<Integer>(); for (int i : nums) { if (pq.size() < k) { pq.add(i); } else if (i > pq.peek()) { pq.poll(); pq.add(i); } } return pq.peek(); } }
posted on 2020-01-17 15:00 IncredibleThings 阅读(125) 评论(0) 编辑 收藏 举报