Quick-selection 选出第k小

目标
给定一个数组,选出第k小的元素
建议参考 快速排序

算法描述

1. 洗牌, 将数组随机打乱
2. 选定一个key值进行划分,<key到key左边,>key到key的右边
3. 直到在划分的过程中找到第k大, 或者完成升序排序之后返回下标为k的元素值

Java实现

import java.util.Arrays;
import java.util.Collections;
import java.util.List;

public class Select {
    /*选择*/
    public static Comparable select(Comparable[] a, int k1) {
        int k = k1 - 1;
        // 洗牌
        List<Comparable> list = Arrays.asList(a);
        Collections.shuffle(list);
        Comparable[] arr = (Comparable[]) list.toArray();
        // 选择
        int lo = 0, hi = arr.length - 1;
        while (hi > lo) {
            int j = partition(a, lo, hi);
            if (j < k) { // 第k小在 j 之后
                lo = j + 1;
            } else if (j > k) { // 第k小在 j 之前
                hi = j - 1;
            } else {
                // 划分的过程中找到kth小
                return a[k];
            }
        }
        // 排序结束后找到kth小
        return a[k];
    }

    /*划分*/
    private static int partition(Comparable[] a, int lo, int hi) {
        Comparable key = a[lo];
        int i = lo + 1, j = hi;
        while (true) {
            while (key.compareTo(a[i]) > 0 && i < hi) {
                i++;
            }
            while (key.compareTo(a[j]) < 0) {
                j--;
            }
            if (i >= j) {
                break;
            }
            exch(a, i, j);
        }
        exch(a, lo, j);
        return j;
    }

    /*交换*/
    private static void exch(Comparable[] a, int i, int j) {
        Comparable tmp = a[i];
        a[i] = a[j];
        a[j] = tmp;
    }

    /*测试*/
    public static void main(String[] args) {
        Integer[] a = new Integer[]{2, 4, 5, 2, 3, 8, 9, 87, 22, 45, 10, 1, 6, 99};
        Integer[] b = a.clone();
        System.out.println(Arrays.toString(a));
        Comparable k = Select.select(a, 14);
        Arrays.sort(b);
        System.out.println(Arrays.toString(b));
        System.out.println(k);
        System.out.println("end");
    }
}

算法分析
时间复杂度: 平均O(n) 最差O(1/2 * n^2)