码迷,mamicode.com
首页 > 其他好文 > 详细

自定义索引优先队列

时间:2021-01-08 11:45:21      阅读:0      评论:0      收藏:0      [点我收藏+]

标签:定位   turn   sem   com   最大   ++   删除索引   extends   --   

/**
 * @desc: 自定义索引优先队列,可修改,删除索引对应的值
 * @author: 毛会懂
 * @create: 2021-01-06 17:20:00
 **/
public class MyIndexPriorityQueue<T extends Comparable<T>> {
    private T[] arr;//存储数组元素
    private int[] pq;//存储元素的下标,符合堆结构
    private int[] qp;//存储元素下标的堆的反序,pq数组中的值作为索引,pq数组的索引作为值j
    private int size;

    public MyIndexPriorityQueue(Integer size){
        arr = (T[])new Comparable[size];
        pq = new int[size + 1];
        qp = new int[size + 1];
        for (int i = 0; i < qp.length; i++) {
            //默认情况下,qp逆序中不保存任何索引
            qp[i] = -1;
        }
        this.size = 0;
    }

    //队列大小
    public Integer getSize(){
        return size;
    }

    //是否为空
    public Boolean isEmpty(){
        return size == 0;
    }

    //某一个位置是否有元素
    public Boolean contains(int k){
        return qp[k] != -1;
    }

    //指定位置,添加元素
    public void add(int i,T t){
        arr[i] = t;
        size++;
        //元素的下标需要符合堆结构
        pq[size] = i;
        qp[i] = size;
        swim(size);
    }

    //删除最大元素
    public T delMax(){
        if(size <= 0){
            return null;
        }
        T t = arr[pq[1]];
        change(1,size);
        size--;
        sink(1);
        return t;
    }

    //修改某一个位置的元素
    public void update(int k ,T t){
        if(!contains(k)){
            return;
        }
        //修改元素值
        arr[k] = t;
        //找到此元素在pq的下标
        int index = qp[k];
        //上浮
        swim(index);
        //下沉
        sink(index);
    }

    //删除某一个位置元素
    public void delete(int k){
        if(!contains(k)){
            return;
        }
        //删除元素
        arr[k] = null;
        //交换位置
        int index = qp[k];
        change(index,size);
        //删除下标
        qp[pq[size]] = -1;
        pq[size] = -1;
        size--;
        //上浮
        swim(index);
        //下沉
        sink(index);
    }

    //最小元素关联的索引
    public int getMinIndex(){
        if(size >=1) {
            return pq[1];
        }
        return -1;
    }


    //上浮
    private void swim(int k){
        while (k > 1){
            if(great(k,k/2)){
                change(k,k/2);
            }
            k = k /2;
        }
    }

    //下沉
    private void sink(int k){
        while (2 * k < size){
            int max;
            if(2 * k + 1 <= size) {
                if (great(2 * k, 2 * k + 1)) {
                    max = 2 * k;
                } else {
                    max = 2 * k + 1;
                }
            }else{
                max = 2 * k;
            }
            if(great(max,k)){
                change(max,k);
            }
            k = max;
        }
    }

    private Boolean great(int i,int j){
        return arr[pq[i]].compareTo(arr[pq[j]]) > 0;
    }

    private void change(int i,int j){
        int temp = pq[i];
        pq[i] = pq[j];
        pq[j] = temp;
        
        qp[pq[i]] = i;
        qp[pq[j]] = j;
    }
}

测试代码:

//索引优先队列
public static void main(String[] args) {
MyIndexPriorityQueue queue = new MyIndexPriorityQueue(20);
queue.add(0,"A");
queue.add(2,"H");
queue.add(1,"D");
queue.add(3,"O");
queue.add(4,"E");
queue.add(5,"W");
queue.add(6,"G");
queue.add(7,"T");


System.out.println("最大的元素所在的索引:"+queue.getMinIndex());
System.out.println("修改元素");
queue.update(6,"Z");
System.out.println("删除元素");
queue.delete(2);
for(int i = 0;i < 8;i++){
System.out.print(queue.delMax() + ", ");
}
System.out.println("over");
}

自定义索引优先队列

标签:定位   turn   sem   com   最大   ++   删除索引   extends   --   

原文地址:https://www.cnblogs.com/maohuidong/p/14245998.html

(0)
(0)
   
举报
评论 一句话评论(0
登录后才能评论!
© 2014 mamicode.com 版权所有  联系我们:gaon5@hotmail.com
迷上了代码!