数据结构与算法Java篇

求最小的K个数

2018-10-22  本文已影响0人  NetCedar

题目描述
    输入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;
    }

}
上一篇下一篇

猜你喜欢

热点阅读