优先队列的代码实现

优先队列的代码实现

import java.util.Arrays;

/**
 * @author lizz
 * @ClassName PriorityQueue.java
 * @Description 优先队列的代码实现
 *  入队就是堆的插入操作
 *  出队就是删除堆顶节点
 * @createTime 2022年04月20日 16:01:00
 */
public class PriorityQueue {
    private int[] array;
    private int size;
    public PriorityQueue(){
        //队列初始长度为32
        array = new int[32];
    }

    /**
     * 入队
     * @param key 入队元素
     */
    public void enQueue(int key){
        //队列长度超过范围，扩容
        if(size >= array.length){
            resize();
        }
        array[size++] = key;
        upAdjust();
    }

    /**
     * 出队
     * @return 出队元素
     */
    public int deQueue(){
        if(size < 0){
            throw new RuntimeException("队列为空！");
        }
        // 获取堆顶元素
        int head = array[0];
        //让最后一个元素移动到堆顶
        array[0] = array[--size];
        downAdjust();
        return head;
    }

    /**
     * "上浮"调整
     */
    public void upAdjust(){
        int childIndex = size-1;
        int parentIndex = (childIndex-1)/2;
        //用tmp保存插入的叶子节点值，用于最后的赋值
        int tmp = array[childIndex];
        while(childIndex >0 &&  tmp > array[parentIndex]){
            // 无需真正交换，简单赋值即可
            array[childIndex] = array[parentIndex];
            childIndex = parentIndex;
            parentIndex = (parentIndex-1)/2;
        }
        array[childIndex] = tmp;
    }

    /**
     * "下沉"调整
     */
    public void downAdjust(){
        //用tmp保存父节点值，用于最后的赋值
        int parentIndex = 0;
        int tmp = array[parentIndex];
        int childIndex = 1;
        while(childIndex < size  ){
            //如果有右孩子，且右孩子小于左孩子的值，则定位到右孩子
            if(childIndex+1 < size && array[childIndex+1] > array[childIndex]){
                childIndex ++ ;
            }
            //如果父节点大于任何一个子节点 直接跳出
            if(tmp >= array[childIndex]){
                break;
            }
            // 无需真正交换，简单赋值即可
            array[parentIndex] = array[childIndex];
            parentIndex = childIndex;
            childIndex = 2 * childIndex + 1;
        }
        array[parentIndex] = tmp;
    }

    /**
     * 队列扩容
     */
    private void resize(){
        int newSize = this.size * 2;
        this.array = Arrays.copyOf(this.array,newSize);
    }

    public static void main(String[] args) {
        PriorityQueue priorityQueue = new PriorityQueue();
        priorityQueue.enQueue(3);
        priorityQueue.enQueue(5);
        priorityQueue.enQueue(10);
        priorityQueue.enQueue(2);
        priorityQueue.enQueue(7);
        int size = priorityQueue.size;
        while (size>0){
            System.out.println("出队元素："+priorityQueue.deQueue());
            size--;
        }
    }
}