求第k小的元素

用快排解决:

用快排,一趟排序后,根据基准值来缩小问题规模。基准值的下角标i 加1 表示了基准值在数组中第几小。如果k<i+1,那就在左半边找;如果k>i+1那就在右半边找。当基准值的下角标+1=k,那就找到答案了。

public class FindTopKth {
    private FindTopKth() {}

    /**
     * @param arr   待查找的数组
     * @param left  待查找数组的左边界
     * @param right 待查找数组的右边界
     * @param k     查找第k小的元素
     * @param <T>   泛型
     * @return 返回第k小的元素
     */
    public static <T extends Comparable<? super T>> T find(T[] arr, int left, int right, int k) {
        int p = partition(arr, left, right, k);
        if (p == k) {
            return arr[p];
        } else if (k < p) {
            return find(arr, left, p - 1, k);
        } else {
            return find(arr, p + 1, right, k);
        }
    }

    private static <T extends Comparable<? super T>> int partition(T[] arr, int left, int right, int k) {
        T base = arr[left];
        int j = left;
        for (int i = left+1; i <= right; i++) {
            if (arr[i].compareTo(base) < 0) {
                swap(arr, i, ++j);
            }
        }
        swap(arr, left, j);
        return j;
    }

    private static void swap(Object[] arr, int i, int j) {
        if (i != j) {
            Object temp = arr[i];
            arr[i] = arr[j];
            arr[j] = temp;
        }
    }

    public static <T extends Comparable<? super T>> T find(T[] arr, int k) {
        return find(arr,0,arr.length-1,k-1);
    }

    public static void main(String[] args) {
        Integer[] arr = {3, 1, 9, 7, 5, 11};
        System.out.println(find(arr,1));//第1小的 //1
        System.out.println(find(arr,2));//第2小的 //3
        System.out.println(find(arr,3));//第3小的 //5
        System.out.println(find(arr,4));//第4小的 //7
        System.out.println(find(arr,5));//第5小的 //9
        System.out.println(find(arr,6));//第6小的 //11
    }
}

  

posted @ 2017-11-30 17:06  GoldArowana  阅读(711)  评论(0编辑  收藏  举报