第九章 中位数和顺序统计量 9.2 期望为线性时间的选择算法

package chap09_Medians_and_Order_Statistics;



import static org.junit.Assert.*;



import java.util.Random;



import org.junit.Test;



public class SearchAlorithms {

    /**

     * 分割（快速排序中对数组的分割）

     * 

     * @param n

     * @param start

     * @param end

     * @return

     */

    protected static int partition(int[] n, int start, int end) {

        int p = end - 1;

        int s = start;// s位于大于a[p]和小于a[p]之间的位置

        int tmp;

        for (int i = start; i < end; i++) {

            if (n[i] < n[p]) {

                tmp = n[i];

                n[i] = n[s];

                n[s] = tmp;

                s++;

            }

        }

        {

            tmp = n[s];

            n[s] = n[p];

            n[p] = tmp;

        }

        return s;

    }



    /**

     * 随机分割，将一个数组从随机选择的位置处分割，左边小，右边大

     * 

     * @param n

     * @param start

     * @param end

     * @return

     */

    static protected int randomPartition(int[] n, int start, int end) {

        Random rand = new Random(23);

        int t = rand.nextInt(end - start) + start;

        int tmp;

        {

            tmp = n[t];

            n[t] = n[end - 1];

            n[end - 1] = tmp;

        }

        return partition(n, start, end);

    }



    /**

     * 搜索数组n中从start到end之间第k小的数字

     * 

     * @param n

     * @param start

     * @param end

     * @param k

     * @return

     */

    static int randomSearch(int[] n, int start, int end, int k) {

        if (start == end - 1)

            return n[start];

        int q = randomPartition(n, start, end);

        int p = q - start + 1;

        if (k == p)

            return n[q+1];

        else if (k > p)

            return randomSearch(n, q, end, k - q);

        else

            return randomSearch(n, start, q, k);

    }



    @Test

    public void testName() throws Exception {

        int[] n = { 13, 19, 9, 5, 12, 8, 7, 4, 21, 2, 6, 11 };

        int a = randomSearch(n, 0, 12, 1);

        System.out.println(a);

    }



}