擬似コード プログラミングに基づいて、ランダムな位置順序選択アルゴリズムを実装します。

擬似コード プログラミングに基づいて、ランダムな位置順序選択アルゴリズムを実装します。

import java.util.Random;

public class randomizedSelection {
    
    
    public static void main(String[] args){
    
    
        int[] A = {
    
    9,8,7,6,5,4,3,2,1};
        //下面中randomizedSelection(A, 0, A.length-1, 5)里的第四个参数，代表的是寻找该数组中排序后的第几个元素。所以，第四个参数的值不能超过数组的长度
        int result = randomizedSelection(A, 0, A.length-1, 5);
        System.out.println(result);
    }
	//下面的代码跟快速排序的代码一样
    private static int partition(int[] A, int p, int r){
    
    
        int x = A[r];
        int i = p-1;
        for(int j = p; j<=r-1; j++){
    
    
            if(A[j]<=x){
    
    
                int tmp = A[i+1];
                A[i+1] = A[j];
                A[j] = tmp;
                i++;
            }
        }
        int t1 = A[i+1];
        A[i+1] = A[r];
        A[r] = t1;
        return i+1;
    }
	//生成随机数并交换数组的最后一个值，与快速排序中的一样
    private static int randomizedPartition(int[] A, int p, int r){
    
    
        int random = (int) (Math.random() * ((r - p) + 1)) + p;
        int temp = A[random];
        A[random] = A[r];
        A[r] = temp;
        return partition(A, p, r);
    }
	//下面就是与快速排序的区别，判定大小并选择性递归
    private static int randomizedSelection(int[] A, int p, int r, int k){
    
    
        int q = randomizedPartition(A, p, r);
        int x;
        if (k==q-p+1){
    
    
            x = A[q];
        }else if (k<q-p+1){
    
    
            x = randomizedSelection(A, p, q-1, k);
        }else{
    
    
            x=  randomizedSelection(A, q+1, r, k-(q-p+1));
        }
        return x;
    }
}