求最小的K个数

作者: NetCedar | 来源:发表于2018-10-22 11:17 被阅读0次

    题目描述
        输入n个整数,找出其中最小的K个数。例如输入4,5,1,6,2,7,3,8这8个数字,则最小的4个数字是1,2,3,4。
    解题思路
        核心思想是利用快速排序,在快速排序过程中,一次排序后,基准值前面的数字都小于基准值,基准值后面的数都大于基准值。那么就可以利用这个特性,当基准值下标对应的位置大于k的时候,就在基准值左边利用快速排序,否则在右边利用快速排序。

    import java.util.ArrayList;
    public class Solution {
        public static ArrayList<Integer> GetLeastNumbers_Solution(int[] input, int k) {
            //用来存放结构
            ArrayList<Integer> result=new ArrayList<Integer>(k);
            if (input==null){
                return null;
            }
             if (k>input.length||k<=0){
                return result;
            }
            int start=0;
            int end=input.length-1;
    
            int index=partition(input,start,end); //一次快排之后 index是返回的下标值 对应的元素个数
            while (index!=k-1){
                //如果元素比k多,在左边利用快速排序
                if (index>k-1){
                    end=index-1;
                    index=partition(input,start,end);
                }else {
                    //否则在右边利用快速排序
                    start=index+1;
                    index=partition(input,start,end);
                }
            }
            
            //将结构存入链表中
            for (int i=0;i<k;i++){
                result.add(input[i]);
                System.out.println(input[i]);
            }
            return result;
        }
    
        /**
         * 根据快排的改进
         * @param array
         * @param left
         * @param right
         * @return
         */
        public static int partition(int[] array, int left, int right){
            //设置的基准
            int index=array[left];
            if (left>right){
                return -1;
            }
            while (left<right){
                //后面的指针向前查找比基准值小的数
                while (left<right&&array[right]>=index){
                    right--;
                }
                //找到了将后面的值放到前面去
                array[left]=array[right];
    
                //前面的指针向后查找,找到比基准值大的数
                while (left<right&&array[left]<=index){
                    left++;
                }
                array[right]=array[left];
            }
            array[left]=index;  //最后将基准值放到空缺的位置
            return left;
        }
    
    }
    

    相关文章

      网友评论

        本文标题:求最小的K个数

        本文链接:https://www.haomeiwen.com/subject/owiwzftx.html