zoukankan      html  css  js  c++  java
  • 求第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
        }
    }
    

      

    ---------------------------------------------------------
    学如不及,犹恐失之
  • 相关阅读:
    [CAMCOCO][C#]我的系统架构.服务器端.(一)
    开博啦,加油!!
    Django RestFul framework Serializer序列化器的使用(定义)
    Django项目快速搭建
    Django简介
    python搭建虚拟环境
    大专生自学web前端前前后后
    实现资源国际化
    利用ajax实现页面动态修改
    七牛使用
  • 原文地址:https://www.cnblogs.com/noKing/p/7929951.html
Copyright © 2011-2022 走看看