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

  

转载于:https://www.cnblogs.com/noKing/p/7929951.html

  • 0
    点赞
  • 0
    收藏
    觉得还不错? 一键收藏
  • 0
    评论
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值