public class FindMinKth {
public static int minKth2(int[] array, int k){
int[] arr = copyArray(array);
return process2(arr,0,arr.length - 1, k-1);
}
public static int process2(int[] arr, int L, int R, int index){
if (L == R){
return arr[L];
}
int pivot = arr[L + (int)(Math.random() * (R - L + 1))];
int[] range = partition(arr, L, R, pivot);
if(range[0] <= index && index <= range[1]){
return arr[index];
}else if(index < range[0]){
return process2(arr, L, range[0]-1, index);
}else{
return process2(arr,range[1]+1, R, index);
}
}
public static int[] partition(int[] arr, int L, int R, int pivot){
int less = L - 1;
int cur = L;
int more = R + 1;
while(cur < more){
if(arr[cur] < pivot){
swap(arr, ++less, cur++);
}else if(arr[cur] > pivot){
swap(arr,cur, --more);
}else{
cur++;
}
}
return new int[]{less+1, more-1};
}
public static void swap(int[] arr, int i1, int i2){
int temp = arr[i2];
arr[i2] = arr[i1];
arr[i1] = temp;
}
public static int[] copyArray(int[] arr){
int[] ans = new int[arr.length];
for (int i = 0; i < ans.length; i++) {
ans[i] = arr[i];
}
return ans;
}
public static int[] generateRandomArray(int maxSize, int maxValue){
int[] arr = new int[(int)(Math.random()*maxSize) + 1];
for (int i = 0; i < arr.length; i++) {
arr[i] = (int)(Math.random() * (maxValue + 1));
}
return arr;
}
public static void main(String[] args) {
int testTime = 50;
int maxSize = 20;
int maxValue = 100;
for (int i = 0; i < testTime; i++) {
int[] arr = generateRandomArray(maxSize,maxValue);
for (int j = 0; j < arr.length; j++) {
System.out.print(arr[j]+" ");
}
System.out.println();
int k = (int)(Math.random() * arr.length)+1;
System.out.println(minKth2(arr,k));
System.out.println("====================="+k);
}
}
}
数据结构-在无序数组中找到第k小的数
最新推荐文章于 2024-10-05 15:32:05 发布