騰訊// 陣列中的第K個最大元素
阿新 • • 發佈:2018-12-25
在未排序的陣列中找到第 k 個最大的元素。請注意,你需要找的是陣列排序後的第 k 個最大的元素,而不是第 k 個不同的元素。
示例 1:
輸入: [3,2,1,5,6,4] 和
k = 2
輸出: 5
示例 2:
輸入: [3,2,3,1,2,4,5,5,6] 和
k = 4
輸出: 4
說明:
你可以假設 k 總是有效的,且 1 ≤ k ≤ 陣列的長度。
class Solution { public int findKthLargest(int[] nums, int k) { Arrays.sort(nums); int len = nums.length; return nums[len-k]; } }
堆排序:
class Solution { public int findKthLargest(int[] nums, int k) { PriorityQueue<Integer> pq = new PriorityQueue<Integer>(); for(int var:nums){ pq.add(var); if(pq.size()>k){ pq.poll(); } } return pq.peek(); } }