题目描述:
在未排序的数组中找到第 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) {
int []copy = Arrays.copyOf(nums, nums.length);
Arrays.sort(copy);
return copy[nums.length - k];
}
}
但是没有什么技术含量,排名靠前的代码
class Solution {
public int findKthLargest(int[] nums, int k) {
int left=0,right=nums.length-1;
while(left<right){
int pivot = partition(nums, left,right);
if(pivot ==k-1) {
return nums[pivot];
} else if(pivot>k-1){
right=pivot-1;
} else {
left=pivot+1;
}
}
return nums[left];
}
private int partition(int[] nums,int left,int right) {
//先获取三个数的中位数
int pivot = median3(nums,left,right);
//int pivot = nums[left];
int start=left,end=right-1;
while(start<end) {
//从pivot左边找起,停在第一个比pivot小的地方,等待交换
while(nums[++start]>pivot) {}
//从pivot右边朝气,停在第一个比pivot大的地方,等待交换
while(nums[--end]<pivot) {}
if(start<end) {
swap(nums,start,end);
}
}
//此时,交换start与pivot
swap(nums,start, right-1);
return start;
}
private int median3(int[] nums,int left, int right){
int median=(left+right)/2;
if(nums[left]<nums[median]) {
swap(nums, left, median);
}
if(nums[left]<nums[right]) {
swap(nums,left, right);
}
if(nums[median]<nums[right]) {
swap(nums, median, right);
}
swap(nums, median, right-1);
return nums[right-1];
}
private void swap(int[] nums, int left, int right) {
int temp = nums[left];
nums[left] = nums[right];
nums[right] = temp;
}
}