求第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
    }
}

  

优质内容筛选与推荐>>
1、[bzoj2115][Wc2011] Xor
2、DataSet
3、公司oracle数据库的一些处理
4、Matrix: android 中的Matrix (android.graphics.Matrix) (转)
5、java数组越界错误


长按二维码向我转账

受苹果公司新规定影响,微信 iOS 版的赞赏功能被关闭,可通过二维码转账支持公众号。

    阅读
    好看
    已推荐到看一看
    你的朋友可以在“发现”-“看一看”看到你认为好看的文章。
    已取消,“好看”想法已同步删除
    已推荐到看一看 和朋友分享想法
    最多200字,当前共 发送

    已发送

    朋友将在看一看看到

    确定
    分享你的想法...
    取消

    分享想法到看一看

    确定
    最多200字,当前共

    发送中

    网络异常,请稍后重试

    微信扫一扫
    关注该公众号