----------------------------------------------本题链接----------------------------------------------
题目描述
有一个整数数组,请你根据快速排序的思路,找出数组中第K大的数。
给定一个整数数组a,同时给定它的大小n和要找的K(K在1到n之间),请返回第K大的数,保证答案存在。
示例
输入
[1,3,5,2,2],5,3
返回值
2
思路
题目说的很清楚了,运用快速排序的思想,这里再给出快速排序代码:
该题可见[牛客题霸-研发] NC4-排序
public class Solution {
/**
* 代码中的类名、方法名、参数名已经指定,请勿修改,直接返回方法规定的值即可
* 将给定数组排序
* @param arr int整型一维数组 待排序的数组
* @return int整型一维数组
*/
public int[] MySort (int[] arr) {
if(arr.length == 0) return null;
quickSort(arr, 0, arr.length-1);
return arr;
}
private void quickSort(int[] arr, int left, int right){
if(left >= right) return;
int point = partition(arr, left, right);
quickSort(arr, left, point-1);
quickSort(arr, point+1, right);
}
private int partition(int[] arr, int left, int right){
int first = arr[left];
while(left < right){
while(left < right && arr[right] >= first) right--;
arr[left] = arr[right];
// swap(arr, left, right);
while(left < right && arr[left] <= first) left++;
arr[right] = arr[left];
// swap(arr, left, right);
}
arr[left] = first;
return left;
}
private void swap(int[] arr, int i, int j){
int tmp = arr[i];
arr[i] = arr[j];
arr[j] = tmp;
}
}
算法过程
- 判断异常情况
- 快速排序找到中轴pivot等于K的位置
- 输出结果
解答
public class Solution {
public int findKth(int[] a, int n, int K) {
return findK(a, 0, n-1, K);
}
private int partition(int[] arr, int left, int right){
int pivot = arr[left];
while(left < right){
while(left < right && arr[right] <= pivot) right--;
arr[left] = arr[right];
while(left < right && arr[left] >= pivot) left++;
arr[right] = arr[left];
}
arr[left] = pivot;
return left;
}
private int findK(int[] arr, int left, int right, int k){
if(left > right) return -1;
int pivot = partition(arr, left, right);
if(pivot == k-1) return arr[pivot];
else if(pivot < k - 1) return findK(arr, pivot+1, right, k);
else return findK(arr, left, pivot-1, k);
}
}