java找出大数据中的 TOP K 问题

    public static void main(String[]args){
        Heap heap = new Heap(10);
        Random r = new Random();
        for (int i = 0;i<10000000;i++){
            heap.add(r.nextInt(1000000000));
        };
        heap.sort();
        System.out.println(Arrays.toString(heap.heap));
    }
    //堆根即arr[0] 两子节点arr[2*0+1],arr[2*0+2], 同理 arr[1]的两子节点 arr[1*2+1],arr[1*2+2] ....
    //小根堆(arr[i]<arr[2i+1] && arr[i]<arr[2i+2])(父节点小于子节点) 用来实现 TOP K 功能
    //1.维护一个大小为 K 的堆
    //2.插入元素时,将大于根的元素放入根的位置,并进行一次交换,保持堆的性质
    static class Heap{
        int [] heap;//用来存储数据
        public Heap(int size){//设置堆大小
            heap = new int[size];//初始化数组
        }
        public void add(int data){//添加元素
            if (data>heap[0]){//由于是小根堆,判断当前元素是否比最小的大
                heap[0]=data;//如果大替换为
                minToRoot(heap.length);//在当前数组大小范围内进行交换
            }
        }
        private void minToRoot(int k){//指定数组下标范围内交换
            if (k<2) return;
            for (int i = 0; 2*i+1<k; ){//有子节点
                //有右节点,且右节点小于左节点 和 父节点
                if (2*i+2<k && heap[2*i+2]<heap[2*i+1] && heap[2*i+2]<heap[i]){
                    swap(i,2*i+2);//交换
                    i = 2*i+2;
                }else if (heap[2*i+1]<heap[i]){//左节点小于父节点
                    swap(i,2*i+1);//交换
                    i = 2*i+1;
                }else{ //当前节点就是最小的,不用向下进行比较了
                    break;
                }

            }
        }
        private void swap(int a,int b){//交换数组中两个下标的值
            int t = heap[a];
            heap[a] = heap[b];
            heap[b] = t;
        }
        public void  sort(){//堆排序  1.最小值->arr[len-1] 2.排序 3.最小值->arr[len-2] .....
            int k = heap.length;
            while (k-1>0){
                swap(0,k-1);//当前堆根最小 将堆根与(k的范围内)最后的元素交换
                k--;
                minToRoot(k);//在下标k的返回内 保持堆性质
            }
        }
    }