目录
寻找第K大
题目描述
有一个整数数组,请你根据快速排序的思路,找出数组中第K大的数。
给定一个整数数组a,同时给定它的大小n和要找的K(K在1到n之间),请返回第K大的数,保证答案存在。
示例1
输入
[1,3,5,2,2],5,3
返回值
2
方法:快速排序
思路就是快速排序的划分,每次划分都会最终确定一个元素的位置,
- 如果这个位置正好是第K大数的位置,那么就找到了
- 如果小于第K大数的位置,则在右边继续寻找
- 否则,在左边继续寻找
import java.util.*;
public class Solution {
public int findKth(int[] a, int n, int K) {
// write code here
int res=QuickSort(a,0,n-1,K);
return res;
}
public static int QuickSort(int[] arr,int low,int high,int K){
if(low<high){
int partition=Partition(arr,low,high);
if(K-1==partition) {
return arr[partition];
}else if(K-1<partition){
QuickSort(arr,low,partition-1,K);
}else {
QuickSort(arr, partition + 1, high, K);
}
}
return arr[K-1];
}
public static int Partition(int[] arr,int low,int high){//从大到小排列
int curVal=arr[low];
int curPos=low;
for (int i = low+1; i <= high; i++) {
if(arr[i]>curVal){
curPos++;
if(curPos!=i){
swap(arr,curPos,i);
}
}
}
arr[low]=arr[curPos];
arr[curPos]=curVal;
return curPos;
}
public static void swap(int[] arr,int i,int j){
int temp=arr[i];
arr[i]=arr[j];
arr[j]=temp;
}
}
快排的时间和空间复杂度
方法二:堆排序
为了巩固之前学习的知识,写了一遍堆排序
import java.util.*;
public class Solution {
public int findKth(int[] a, int n, int K) {
// write code here
int res=heapSort(a,K);
return res;
}
//堆排序
public static int heapSort(int[] arr,int K){
for (int i = arr.length/2-1; i >=0 ; i--) {//从第一个非叶子节点开始调整
adjustHeap(arr,i,arr.length);
}
for (int j = 1; j < K; j++) {
swap(arr,0,arr.length-j);
adjustHeap(arr,0,arr.length-j);
}
return arr[0];
}
public static void adjustHeap(int[] arr,int i,int length){
int temp=arr[i];
for (int j = 2*i+1; j < length; j++) {
if(j+1<length&&arr[j]<arr[j+1]){
j++;
}
if(arr[j]>temp){
swap(arr,i,j);
i=j;
}
}
}
public static void swap(int[] arr,int i,int j){
int temp=arr[i];
arr[i]=arr[j];
arr[j]=temp;
}
}
方法三:归并排序
巩固排序算法知识
import java.util.*;
public class Solution {
public int findKth(int[] a, int n, int K) {
// write code here
mergeSort(a,0,a.length-1);
return a[K-1];
}
//归并排序
public static void mergeSort(int[] arr,int low,int high){
if(low<high){
int mid=(low+high)/2;
mergeSort(arr,low,mid);
mergeSort(arr,mid+1,high);
merge(arr,low,mid,high);
}
}
public static void merge(int[] arr,int low,int mid,int high){
int[] temp=new int[arr.length];
for (int i = 0; i < arr.length; i++) {
temp[i]=arr[i];
}
int i=low;
int j=mid+1;
int k=low;
while(i<=mid&&j<=high){
if(temp[i]>temp[j]){
arr[k++]=temp[i++];
}else{
arr[k++]=temp[j++];
}
}
while(i<=mid){
arr[k++]=temp[i++];
}
while(j<=high){
arr[k++]=temp[j++];
}
}
}