算法练习篇之：最小的k个数（链表）

题目描述

输入n个整数，找出其中最小的K个数。例如输入4,5,1,6,2,7,3,8这8个数字，则最小的4个数字是1,2,3,4。

解题思路

法1：先对数组排序，然后取出前k个
法2：利用最大堆保存这k个数，每次只和堆顶比，如果比堆顶小，删除堆顶，新数入堆。

方法一：思路十分简单，只需将数组进行排序即可，排序的方法快排，归并都是O（nlogn）复杂度的算法，归并排序需要额外的内存空间，快排不是稳定排序方法。
排序后，即可将其依次插入到arraylist中去即可！

方法二：需要借助堆的数据结构，通过比较器构造，每次取出的都是堆中最大元素的值。在{4,5,1,6,2,7,3,8}中，需要取最小的4个值，首先依次向堆中插入四个数值，然后每次再从堆中，查看堆顶的元素的下一个数组中的元素比较大小，如果堆顶元素（堆中最大的值）比数组中的值大时，将堆中元素弹出，数组元素插入，否则，继续读取数组的下一个元素，然后重复上面的对比！最后留在堆中的四个元素即为最小的4个**，最后依次将其添加到arraylist中**去！

代码实现

去博客设置页面，选择一款你喜欢的代码片高亮样式，下面展示同样高亮的 代码片.

import java.util.ArrayList;
import java.util.Arrays;
import java.util.Comparator;
import java.util.PriorityQueue;

public class getLeastNumbers {
    //获取最小的k个数
    public ArrayList<Integer> getLeast_2(int[] num,int k){
        ArrayList<Integer> res=new ArrayList<>();
        if(num==null||k==0||k>num.length){
            return res;
        }
        PriorityQueue<Integer> maxHeap=new PriorityQueue<>(k, new Comparator<Integer>() {
            @Override
            public int compare(Integer o1, Integer o2) {
                return o2-o1;
            }
        });//实现大顶堆
        for (int i=0;i<num.length;i++){
            if (maxHeap.size()!=k){//先向堆中插入k个元素
                maxHeap.offer(num[i]);
            }else{//之后每次弹出堆中最大的元素进行遍历比较，然后用小的元素将堆中大的元素替换掉
                if (maxHeap.peek()>num[i]){
                    maxHeap.poll();
                    maxHeap.offer(num[i]);
                }
            }
        }
        for (Integer i:maxHeap){
            res.add(i);
        }
        return res;
    }
    public ArrayList<Integer> getLeast_1(int[] num,int k){//方法2比较简单，直接将数组元素进项排序即可
        ArrayList<Integer> res=new ArrayList<>();
        if (num==null||k==0||k>num.length){
            return res;
        }
        Arrays.sort(num);
        for (int i=0;i<k;i++){
            res.add(num[i]);
        }
        return res;
    }

    public static void main(String[] args) {
        getLeastNumbers test=new getLeastNumbers();
        int[] num={4,5,1,6,2,7,3,8};int k=4;
        ArrayList<Integer> res=test.getLeast_2(num,k);
        System.out.println("最小的"+k+"个值为："+res);
    }
}

总结

本题来源于面试经典教材《剑指offer》中 归属于Heap&排序类型题目。
同许多在算法道路上不断前行的人一样，不断练习，修炼自己！
如有博客中存在的疑问或者建议，可以在下方留言一起交流，感谢各位！
最后，感谢AIAS！