链接: 215
在未排序的数组中找到第 k 个最大的元素。请注意,它是数组有序排列后的第 k 个最大元素,而不是第 k 个不同元素。
例如,
给出 [3,2,1,5,6,4]
和 k = 2,返回 5。
注意事项:
你可以假设 k 总是有效的,1 ≤ k ≤ 数组的长度。
这个题可以用优先队列来解决, 用java提供的类库会非常省事.
public int findKthLargest(int[] nums, int k) { if(k > nums.length || k < 1) return Integer.MIN_VALUE; PriorityQueue<Integer> pq = new PriorityQueue<>(k); for(int i=0;i<k;i++) { pq.add(nums[i]); } for(int i=k;i<nums.length;i++) { if(nums[i] > pq.peek()) { pq.poll(); pq.add(nums[i]); } } return pq.peek(); }
另外也可以用堆来解决这个问题:
public class Leet215FindKthLargest { public int findKthLargest(int[] nums, int k) { if(k > nums.length || k < 1) return Integer.MIN_VALUE; int[] minHeap = new int[k+1]; for(int i=1;i<=k;i++) { minHeap[i] = nums[i-1]; adjustHeap(minHeap,i); } for(int i=k+1;i<=nums.length;i++) { int cur = nums[i-1]; if(cur > minHeap[1]) { minHeap[1] = cur; adjustHeapAtHead(minHeap); } } return minHeap[1]; } private void adjustHeap(int[] arr,int index){ while(index > 1) { if(arr[index] < arr[index/2]) { swap(arr,index,index/2); } else { break; } index /= 2; } } private void adjustHeapAtHead(int[] arr) { int index = 1,tmp; while(index <= arr.length-1) { if(index*2 >= arr.length-1) { if(index*2 == arr.length-1 && arr[index] > arr[index*2]) { swap(arr,index,index*2); } break; } else { tmp = arr[index*2] < arr[index*2 +1] ? index*2 : index*2+1; if(arr[tmp] > arr[index]) break; swap(arr,index,tmp); index = tmp; } } } private void swap(int[] arr,int i,int j) { if(i==j) return; arr[i] = arr[i] ^ arr[j]; arr[j] = arr[i] ^ arr[j]; arr[i] = arr[i] ^ arr[j]; } public static void main(String[] args) { Leet215FindKthLargest clazz = new Leet215FindKthLargest(); int[] a = {3,2,3,1,2,4,5,5,6,7,7,8,2,3,1,1,1,10,11,5,6,2,4,7,8,5,6}; clazz.findKthLargest(a,20); System.out.println(Arrays.toString(a)); } }