快排思想,选取序列的一个key进行划分,小于key的分在左边,大于key的在右边,key的位置-low+1就是最后一个元素是key的序列中元素的数量,当元素数量大于K时,就在左半部分递归找,等于时 arr[key]就是第K
大的元素,小于K时,在右边递归找第k-num大的元素
/**
* 文件名:FindK.java
* 时间:2014年11月7日上午11:23:43
* 作者:修维康
*/
package chapter7;
/**
* 类名:FindK 说明:找到一个数组中第K大的元素
*/
public class FindK {
public static int partition(int[] a, int low, int high) {
int key = a[low];
while (low < high) {
while (low < high && a[high] >= key)
high--;
a[low] = a[high];
while (low < high && a[low] <= key)
low++;
a[high] = a[low];
}
a[low] = key;
return low;
}
public static int quickSelect(int[] a, int low, int high, int k) {
if (low == high)
return a[low];
int keyPos = partition(a, low, high);
int num = keyPos - low + 1;
if (num == k)
return a[keyPos];
else if (k < num)
return quickSelect(a, low, keyPos - 1, k);
else
return quickSelect(a, keyPos + 1, high, k - num);
}
/**
* 方法名:main 说明:测试
*/
public static void main(String[] args) {
// TODO Auto-generated method stub
int[] a = new int[] {1,2,3,4,5,6,7,8,9,10};
System.out.println(quickSelect(a, 0, a.length - 1,5));
}
}