优先级队列(堆)的模拟实现

import java.util.Arrays;
public class Test {

    public int[] elem;
    public int usedSize;
    public Test() {
        this.elem = new int[10];
    }

    /**
     * 填充数据
     * @param array
     */
    public void initElem(int[] array) {
        for(int i = 0;i < array.length;i++) {
            elem[i] = array[i];
            usedSize++;
        }
    }

    public void createBigHeap() {
        for(int parent = (usedSize-1-1)/2;parent >= 0;parent--) {
            shiftDown(parent,usedSize);
        }
    }
    /**
     * 向下调整
     * @param
     * @param parent
     */
    public void shiftDown(int parent,int end) {
        int child = 2*parent + 1;

        while(child < end) {
            //如果右孩子存在,找到左右孩子中较小的孩子,用child进行标记
            if(child+1 < usedSize && elem[child] < elem[child+1]) {
                child++;
            }
            //child一定是 左右孩子最大值的下标
            if(elem[child] > elem[parent]) {
                //交换
                swap(child,parent);
                parent = child;
                child = 2*parent+1;
            }else {
                break;
            }
        }
    }

    private void swap(int i,int j) {
        int tmp = elem[i];
        elem[i] = elem[j];
        elem[j] = tmp;
    }

    public int poll() {
        int tmp = elem[0];//删除嘴高的那个值
        swap(0,usedSize-1);
        usedSize--;
        shiftDown(0,usedSize);
        return tmp;
    }


    
    public void offer(int val) {
        //1.判断满
        if(isFull()) {
            //扩容
            this.elem = Arrays.copyOf(elem,2*elem.length);
        }
        //2.插入元素
        elem[usedSize] = val;
        usedSize++;
        //3.开始向上调整
        siftUp(usedSize-1);
    }

    private void siftUp(int child) {
        int parent = (child-1) / 2;
        while(child > 0) {
            if(elem[child] > elem[parent]) {
                swap(child,parent);
                child = parent;
                parent = (child-1)/2;
            }else {
                break;
            }
        }
    }

    public boolean isFull() {
        return usedSize == elem.length;
    }

    public void heapSort() {
        int end = usedSize-1;
        while(end > 0) {
            swap(0,end);
            shiftDown(0,end-1);
            end--;
        }
    }
}

  • 4
    点赞
  • 0
    收藏
    觉得还不错? 一键收藏
  • 0
    评论
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值