TopN问题解决

大家好,好久没更新博客，也是因为最近一直在学习SpringBoot（因为我工作两年了，不是科班出身，不想以后总是这个样子，现在一直在学习springBoot,springCloud，java内存模型以及数据库的优化。也在码云上写几个小的开源小项目。个人感觉自我有了一定的提升。如果朋友们需要什么技术讲解可以在下面留言，我看到了一定会写相应的博客）。在几个月的学习中，发现以前写的东西好幼稚（很多小伙伴都会有看自己以前写的东西幼稚的感觉吧），同时也鼓励大家多自我提升，各种零碎时间利用起来学习学习。

那么言归正传：本文通过小顶堆的方式实现topN问题。大概逻辑就是先在数组的前n个中实现小顶堆（顶最小），后面每一个跟顶比较，如果小于等于顶，pass掉，负责替换顶，并调整最小堆。

代码如下：

public class TopN {

    // 父节点
    private int parent(int n) {
        return (n - 1) / 2;
    }

    // 左孩子
    private int left(int n) {
        return 2 * n + 1;
    }

    // 右孩子
    private int right(int n) {
        return 2 * n + 2;
    }

    // 构建堆
    private void buildHeap(int n, int[] data) {
        for(int i = 1; i < n; i++) {
            int t = i;
            // 调整堆
            while(t != 0 && data[parent(t)] > data[t]) {
                int temp = data[t];
                data[t] = data[parent(t)];
                data[parent(t)] = temp;
                t = parent(t);
            }
        }
    }

    // 调整data[i]
    private void adjust(int i, int n, int[] data) {
        if(data[i] <= data[0]) {
            return;
        }
        // 置换堆顶
        int temp = data[i];
        data[i] = data[0];
        data[0] = temp;
        // 调整堆顶
        int t = 0;
        while( (left(t) < n && data[t] > data[left(t)])
            || (right(t) < n && data[t] > data[right(t)]) ) {
            if(right(t) < n && data[right(t)] < data[left(t)]) {
                // 右孩子更小，置换右孩子
                temp = data[t];
                data[t] = data[right(t)];
                data[right(t)] = temp;
                t = right(t);
            } else {
                // 否则置换左孩子
                temp = data[t];
                data[t] = data[left(t)];
                data[left(t)] = temp;
                t = left(t);
            }
        }
    }

    // 寻找topN，该方法改变data，将topN排到最前面
    public void findTopN(int n, int[] data) {
        // 先构建n个数的小顶堆
        buildHeap(n, data);
        // n往后的数进行调整
        for(int i = n; i < data.length; i++) {
            adjust(i, n, data);
        }
    }

    // 打印数组
    public void print(int[] data) {
        for(int i = 0; i < data.length; i++) {
            System.out.print(data[i] + " ");
        }
        System.out.println();
    }

}

 

 

public static void main(String[] args) {
        int[] arr1 = {2,1,5,4,3,9,6,7,8,0};
        TopN topN = new TopN();
        topN.print(arr1);
        topN.findTopN(4, arr1);
        topN.print(arr1);
    }