Java數(shù)據(jù)結構之堆(優(yōu)先隊列)的實現(xiàn)

更新時間：2022年05月20日 15:54:36 作者：Carol

堆（優(yōu)先隊列）是一種典型的數(shù)據(jù)結構，其形狀是一棵完全二叉樹，一般用于求解topk問題。本文將利用Java語言實現(xiàn)堆，感興趣的可以學習一下

堆（優(yōu)先隊列）是一種典型的數(shù)據(jù)結構，其形狀是一棵完全二叉樹，一般用于求解topk問題。根據(jù)雙親節(jié)點大于等于孩子節(jié)點或雙親節(jié)點小于等于孩子節(jié)點，可分為大頂堆和小頂堆，本文實現(xiàn)大頂堆。

根據(jù)大頂堆的定義，大頂堆的雙親節(jié)點大于等于其孩子節(jié)點，堆頂元素最大，對于每一個子樹都是一個大頂堆，則從最后一個雙親節(jié)點進行調整為大頂堆，一直到根節(jié)點，則可構建一個大頂堆。

我們這里采用數(shù)組去存儲，以heap={3,2,1,5,6,4}為例，需要一個init(int[] heap)初始化方法，從最后一個雙親節(jié)點開始將heap逐漸調整為大頂堆，其中需要使用到adjust(int[] heap, int i, int end)方法。

調整過程：從最后一個雙親節(jié)點出發(fā)，如果以當前雙親節(jié)點為根的樹不符合大頂堆，則進行調整。

代碼實現(xiàn)如下：

public void init(int[] heap) {
        //從最后一個雙親節(jié)點開始調整
        //逐漸往上進行調整
        for (int i = heap.length / 2 ; i > 0 ; i-- ) {
            this.adjust(heap, i, heap.length);
        }
    }

    public void adjust(int[] heap, int i, int end) {
        int j = i << 1;
        while (j <= end) {
            //找到兩個孩子節(jié)點z中較大的節(jié)點
            if (j < end && heap[j - 1] < heap[j]) {
                j = j + 1;
            }
            //如果較大節(jié)點還小于根節(jié)點，則以當前節(jié)點為根節(jié)點的
            //二叉樹已經(jīng)是大頂堆，不需要進行調整
            if (heap[i - 1] > heap[j - 1]) {
                break;
            }
            //進行調整，將當前節(jié)點換到較大位置，再從當前位置進行調整
            int temp = heap[i - 1];
            heap[i - 1] = heap[j - 1];
            heap[j - 1] = temp;
            i = j;
            j = i << 1;
        }
    }

構建好了大頂堆之后，我們如何求得topk呢，此時堆頂元素為top1，我們只需要將top1元素拿走，將剩下元素調整為大頂堆，k次之后即可得到topk。

具體過程：我們將堆頂元素與最后一個元素進行交換，然后將堆頂?shù)降箶?shù)第二個元素進行調整，依次類推。

以leetcode215數(shù)組中第k個最大元素為例：

給定整數(shù)數(shù)組 nums 和整數(shù) k，請返回數(shù)組中第 k 個最大的元素。

請注意，你需要找的是數(shù)組排序后的第 k 個最大的元素，而不是第 k 個不同的元素

public int findKthLargest(int[] nums, int k) {
        this.init(nums);
        //找到第k大的數(shù)
        int end = nums.length;
        while (k > 1) {
            //將當前堆頂元素放到末尾，進行堆調整
            int temp = nums[0];
             nums[0] = nums[end - 1];
             nums[end - 1] = temp;
             end = end - 1;
             -- k;
             this.adjust(nums, 1, end);
        }
        return nums[0];
    }

此外，Java本身提供了優(yōu)先隊列集合類，但是對于這個題目效率不如自己實現(xiàn)的高

public int findKthLargest(int[] nums, int k) {
        PriorityQueue<Integer> priorityQueue = new PriorityQueue<>(k);
        for (int num : nums) {
            if (priorityQueue.size() == k) {
                if (num > priorityQueue.peek()) {
                    priorityQueue.poll();
                    priorityQueue.add(num);
                }
                continue;
            }
            priorityQueue.add(num);
        }
        return priorityQueue.poll();
    }