找出数组中的第K大的元素

问题描述

代码

public class FindK {
//随机取一个基准数pivot，然后遍历数组一次，比基准数大的逐一的放在最前面，最后返回基准数的位置
//并且次基准数在a[left...right]中是第(k-left+1)大的元素
public static int randomPartition(int[] a,int left,int right){
int pivot=a[right],k=left-1;
for(int i=left;i<=right;i++){
if(a[i]>pivot){
int temp=a[i];
a[i]=a[++k];
a[k]=temp;
}
}

int t=a[right];
a[right]=a[++k];
a[k]=t;

return k;
}
//得到第K大的元素
public static int getMaxK(int[] a,int k,int left,int right){
int maxK=randomPartition(a,left,right);
//num表示是第a[maxK]是第num大
int num=maxK-left+1;
if(num==k)
return a[maxK];
else if(num<k)
return getMaxK(a,k-num,maxK+1,right);
else
return getMaxK(a,k,left,maxK-1);
}
//验证
public static void main(String[] args) {
int[] a={12012, 3, 945, 965, 66, 232, 65, 7, 8, 898, 56, 878, 170, 13, 5};

//依次求出第i大的元素最后打印的结果是降序排列
for(int i=0;i<a.length;i++){
System.out.println(getMaxK(a,i+1,0,14));
}
}
}