java线性时间寻找第k小元素_给定线性序集中的n个元素和一个整数k,1≤k≤n,要求找出这n个元素中第k小的元素。-优快云博客

本文链接：https://blog.youkuaiyun.com/ustczyy/article/details/16901461

本文介绍了一个使用Java实现的快速选择算法，用于查找数组中的第k小和第k大元素。通过随机分区策略优化性能，适用于大数据集的高效筛选。

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >

package contcurrentandalgorithm;

import java.util.Random;

/**
*
* @author Administrator
* zyyjiao@mail.ustc.edu.cn
*/
public class SelectKMinAndMax {

    public static void main(String[] args) {
        int a[] = {2, 3, 4, 1, 5, 10, 9, 7, 8, 6};
        int k = 4;
        int result = selectKMin(a, 0, 9, k);
        //SelectKMinAndMax sl = new SelectKMinAndMax();
        //System.out.println(SelectKMinAndMax.selectKMin(a, 0, 9, k));
        System.out.println(result);
    }

    public static void swap(int a[], int i, int j) {
        int temp;
        temp = a[i];
        a[i] = a[j];
        a[j] = temp;
    }

    public static int partion(int a[], int p, int r) {//根据pivot a[r]来划分数组
        int pivot = a[r];
        int low = p - 1;
        int i;
        for (i = p; i < r; i++) {
            if (a[i] < pivot) {
                swap(a, ++low, i);
            }
        }
        swap(a, ++low, r);
        return low;
    }

    public static int RondomPartition(int a[], int p, int r) {
        int i;//随机选择一个pivot来划分
        Random rd = new Random();
        // i = p+(rd)%(r-p+1);
        i = rd.nextInt(10);
        swap(a, i, r);
        return partion(a, p, r);//返回轴位置
    }

    public static int selectKMin(int a[], int p, int r, int k) {
        if (p == r) {
            return a[p];
        }
        int q = RondomPartition(a, p, r);
        int count = q - p + 1;//计算 a[p..q]的元素数量
        if (k == count)//刚好,返回
        {
            return a[q];
        } else if (k < count)//在前半部分
        {
            return selectKMin(a, p, q - 1, k);
        } else //在后半部分
        {
            return selectKMin(a, q + 1, r, k - count);
        }

}
}