海量数据获取TopK、堆排序，快速排序实现

最新推荐文章于 2024-07-17 08:04:18 发布

anhooo

最新推荐文章于 2024-07-17 08:04:18 发布

阅读量451

点赞数

分类专栏：算法文章标签： java 堆排序快速排序算法

本文链接：https://blog.csdn.net/anhooo/article/details/106970801

版权

本文介绍了如何使用Java的PriorityQueue实现海量数据获取最小TopK，以及手写堆排序算法。同时，文章还探讨了快速排序在解决此类问题上的应用。

摘要由CSDN通过智能技术生成

注意：最小TopK用大顶堆，最大TopK用小顶堆

一、用java 的priorityQueue实现

//最小Top， 用java的PriorityQueue
import java.util.*;
public class Solution{
    public ArrayList<Integer<> getLeastNumber_Solution(int[] input, int k){
        int n = input.length;
        ArrayList<Integer> list = new ArrayList<>();
        if(n == 0 || k > n || k == 0) return list;
        //大顶堆 降序
        PriorityQueue<Integer> maxHeap = new PriorityQueue<>(k, new Comparator<Integer>(){
            @Override
            public int compare(Integer o1, Integer o2){
                return o2.compareTo(o1);
            }
        });
        //PriorityQueue<Integer> maxHeap = new PriorityQueue<>(k, (o1, o2) -> o2 - o1);
        for(int i = 0; i < n; i++){
            if(maxHeap.size() < k){
                maxHeap.offer(input[i]);
            }else if(maxHeap.peek