求第k小的元素

时间:2023-03-08 18:35:51
求第k小的元素

用快排解决:

用快排,一趟排序后,根据基准值来缩小问题规模。基准值的下角标i 加1 表示了基准值在数组中第几小。如果k<i+1,那就在左半边找;如果k>i+1那就在右半边找。当基准值的下角标+1=k,那就找到答案了。

public class FindTopKth {
private FindTopKth() {} /**
* @param arr 待查找的数组
* @param left 待查找数组的左边界
* @param right 待查找数组的右边界
* @param k 查找第k小的元素
* @param <T> 泛型
* @return 返回第k小的元素
*/
public static <T extends Comparable<? super T>> T find(T[] arr, int left, int right, int k) {
int p = partition(arr, left, right, k);
if (p == k) {
return arr[p];
} else if (k < p) {
return find(arr, left, p - 1, k);
} else {
return find(arr, p + 1, right, k);
}
} private static <T extends Comparable<? super T>> int partition(T[] arr, int left, int right, int k) {
T base = arr[left];
int j = left;
for (int i = left+1; i <= right; i++) {
if (arr[i].compareTo(base) < 0) {
swap(arr, i, ++j);
}
}
swap(arr, left, j);
return j;
} private static void swap(Object[] arr, int i, int j) {
if (i != j) {
Object temp = arr[i];
arr[i] = arr[j];
arr[j] = temp;
}
} public static <T extends Comparable<? super T>> T find(T[] arr, int k) {
return find(arr,0,arr.length-1,k-1);
} public static void main(String[] args) {
Integer[] arr = {3, 1, 9, 7, 5, 11};
System.out.println(find(arr,1));//第1小的 //1
System.out.println(find(arr,2));//第2小的 //3
System.out.println(find(arr,3));//第3小的 //5
System.out.println(find(arr,4));//第4小的 //7
System.out.println(find(arr,5));//第5小的 //9
System.out.println(find(arr,6));//第6小的 //11
}
}