欢迎您访问程序员文章站本站旨在为大家提供分享程序员计算机编程知识!
您现在的位置是: 首页

java queue-PriorityQueue源码解析

程序员文章站 2024-02-14 23:51:46
...

一、概念

1. 它是基于优先级堆的*优先级队列;

2. 其元素是按照自然排序或者在队列构建时提供的方法来进行排序;

3. 队列中不允许存在null元素;

4. 它不是多线程安全的; 

5. 它使用的是二叉堆,二叉堆分为大根堆和小根堆,大根堆是父节点的值总是大于或等于子节点的值,小根堆是父节点的值总是小于或等于子节点的值;

6. 它总是一棵完全而查实或近似完全二叉树;

7. 它内部的实现是基于数组格式的,对于数组中任意位置n上的元素,它的左孩子是在[2n+1]的位置上,它的右孩子是在[2n+2]的位置上,它的父节点则在[(n-1)/2]上,而根节点在[0]上;


二、PriorityQueue源码分析

1.初始化定义

    在PriorityQueue中初始化定义了5个变量,可以看到它的底层存储逻辑是用数组实现的,默认的初始化容量是11.

//默认容量为11
private static final int DEFAULT_INITIAL_CAPACITY = 11;
//元素的存储方式是数组
private transient Object[] queue;
//队列中元素的个数
private int size = 0;

//比较器
private final Comparator<? super E> comparator;

//队列修改的次数
private transient int modCount = 0;

2.PriorityQueue的构造函数

    PriorityQueue的构造函数有多种,比如下面代码中的前三种都是根据默认容量、默认比较器或者指定的容量或比较器来创建一个空的队列,而在后面几种方法中都是根据指定的集合创建包含指定元素的优先级队列: 

    (1)PriorytyQueue():创建一个默认容量(11)和自然顺序的优先级队列;

    (2)PriorityQueue(int initialCapacity):创建一个指定容量和自然顺序的优先级队列;

    (3)PriorityQueue(int initialCapacity,Comparator<? super E> comparator):创建一个指定容量和指定比较器的优先级队列;

    (4)PriorityQueue(Collection<? extends E> c):创建一个包含指定集合元素的优先级队列,它分为三种情况:

      第一种:如果是一个已经排好序的集合,首先根据集合获取到他的比较器方法,然后将集合中的元素添加到队列中;

      第二种:如果是一个优先级队列,首先获取比较器方法,然后将指定优先级队列中的元素赋值到该队列中;

      第三种:其他情况默认使用自然比较器,将集合中的元素添加到队列中;

    (5)PriorityQueue(PriorityQueue<? extends E> c):创建一个包含指定优先级队列的队列,他的实现逻辑与(4)中的第二种情况相同;

    (6)PriorityQueue(SortedSet<? extends E> c):创建一个包含指定排序集合中元素的优先级队列,他的实现逻辑与(4)中的第一种情况相同

  /**
     * 构造函数:创建一个默认容量(11)和自然顺序的PriorityQueue
     */
    public PriorityQueue() {
        this(DEFAULT_INITIAL_CAPACITY, null);
    }

    /**
     * 构造函数:创建一个指定容量和自然顺序的队列
     * @param initialCapacity
     */
    public PriorityQueue(int initialCapacity) {
        this(initialCapacity, null);
    }

    /**
     * 构造函数:创建一个指定容量和指定顺序的队列
     * @param initialCapacity
     * @param comparator
     */
    public PriorityQueue(int initialCapacity,
                         Comparator<? super E> comparator) {
        if (initialCapacity < 1)
            throw new IllegalArgumentException();
        this.queue = new Object[initialCapacity];
        this.comparator = comparator;
    }

    
    /**
     * 构造函数:根据指定集合创建一个队列
     * 如果c是有序的集合:根据集合获取到对应的比较器comparator,将集合中的元素
     * @param c
     */
    @SuppressWarnings("unchecked")
    public PriorityQueue(Collection<? extends E> c) {
        if (c instanceof SortedSet<?>) {
            SortedSet<? extends E> ss = (SortedSet<? extends E>) c;
            this.comparator = (Comparator<? super E>) ss.comparator();
            initElementsFromCollection(ss);
        }
        else if (c instanceof PriorityQueue<?>) {
            PriorityQueue<? extends E> pq = (PriorityQueue<? extends E>) c;
            this.comparator = (Comparator<? super E>) pq.comparator();
            initFromPriorityQueue(pq);
        }
        else {
            this.comparator = null;
            initFromCollection(c);
        }
    }

    
    /**
     * 构造函数:创建一个指定优先级的集合中的元素的队列
     * @param c
     */
    @SuppressWarnings("unchecked")
    public PriorityQueue(PriorityQueue<? extends E> c) {
        this.comparator = (Comparator<? super E>) c.comparator();
        initFromPriorityQueue(c);
    }

    
    @SuppressWarnings("unchecked")
    public PriorityQueue(SortedSet<? extends E> c) {
        this.comparator = (Comparator<? super E>) c.comparator();
        initElementsFromCollection(c);
    }

    /**
     * 初始化一个队列
     * @param c
     */
    private void initFromPriorityQueue(PriorityQueue<? extends E> c) {
        if (c.getClass() == PriorityQueue.class) {
            this.queue = c.toArray();
            this.size = c.size();
        } else {
            initFromCollection(c);
        }
    }

    /**
     * 从集合中初始化数据到队列中
     * @param c
     */
    private void initElementsFromCollection(Collection<? extends E> c) {
    	//将集合转化成数组
        Object[] a = c.toArray();
        //如果数组不是Object类型,转换成Object类型
        if (a.getClass() != Object[].class)
            a = Arrays.copyOf(a, a.length, Object[].class);
        int len = a.length;
        //如果数组的长度为1或者比较器不为空,如果队列中的元素
        if (len == 1 || this.comparator != null)
            for (int i = 0; i < len; i++)
                if (a[i] == null)
                    throw new NullPointerException();
        this.queue = a;
        this.size = a.length;
    }


    private void initFromCollection(Collection<? extends E> c) {
        initElementsFromCollection(c);
        heapify();
    }
    
    private void heapify() {
        for (int i = (size >>> 1) - 1; i >= 0; i--)
            siftDown(i, (E) queue[i]);
    }

3.PriorityQueue中add方法的实现

    由于PriorityQueue类似一棵完全二叉树,但是它的内部存储结构使用的是数组,并且是*限的,因此在插入一个元素时,可能需要经过多个操作步骤:

    (1)首先需要判断插入的元素是否是null,因为在优先级队列中不允许存在空元素,这是因为在队列中是根据null来判断某个位置是否存在值;

    (2)如果当前队列中的元素的个数大于或等于存储数组的长度,那么需要对数组进行扩容,其调用的方法为grow(minCapacity),其扩容的逻辑是,如果当前数组的长度小于64,那么将扩大一半的长度,否则扩大一倍,同时需要对扩容后的数组的容量进行检查,不能超过最大的限制容量;

    (3)如果当前的队列为空,那么插入的是第一个元素,直接赋值到数组的第[0]位置,结束操作;

    (4)如果当前队列不为空,默认是将新插入的元素添加到数组的尾部,但是由于优先级队列要求及父节点与子节点之间的值有比较的关系,所以可能需要进行调整,主要是在siftUp方法中实现,根据有无比较器决定最终的实现方法,将插入的元素与其父节点元素比较,如果大于其父节点,那么完成插入操作,不需要调整,如果小于其父节点,那么需要与其父节点调换位置,然后在一层层的往上比较,使其满足优先级队列的特性。

  /**
    * 向队列中插入一个元素
    */
    public boolean add(E e) {
        return offer(e);
    }

    /**
     * 向队列中插入一个元素
     * 1.如果元素为空,抛出异常
     */
    public boolean offer(E e) {
        if (e == null)
            throw new NullPointerException();
        modCount++;
        int i = size;
        //如果队列中的元素个数大于或等于队列的长度
        if (i >= queue.length)
            grow(i + 1);
        size = i + 1;
        //如果队列中的元素为空,那么新加入的元素为第一个元素,直接赋值到[0]位置
        if (i == 0)
            queue[0] = e;
        else
            siftUp(i, e);  //插入元素向上进行调整
        return true;
    }
    
    /**
     * 对数组的容量进行扩容
     * @param minCapacity
     */
    private void grow(int minCapacity) {
        int oldCapacity = queue.length;
        //扩容的的大小要么是翻倍要么是增长50%
        int newCapacity = oldCapacity + ((oldCapacity < 64) ?
                                         (oldCapacity + 2) :
                                         (oldCapacity >> 1));
        //当数组的长度超过最大限制时,进行调整
        if (newCapacity - MAX_ARRAY_SIZE > 0)
            newCapacity = hugeCapacity(minCapacity);
        queue = Arrays.copyOf(queue, newCapacity);
    }
    
  //检验数组的长度是否超过限制,如果超过了进行调整或抛出异常
    private static int hugeCapacity(int minCapacity) {
        if (minCapacity < 0) // overflow
            throw new OutOfMemoryError();
        return (minCapacity > MAX_ARRAY_SIZE) ?
            Integer.MAX_VALUE :
            MAX_ARRAY_SIZE;
    }
    
    /**
     * 在k位置插入元素x,同时对数组中的元素进行调整,使其满足队列的要求
     * @param k
     * @param x
     */
    private void siftUp(int k, E x) {
        if (comparator != null)
            siftUpUsingComparator(k, x);
        else
            siftUpComparable(k, x);
    }

    /**
     * 如果是默认比较器,插入元素时的调整方法
     * @param k
     * @param x
     */
    private void siftUpComparable(int k, E x) {
        Comparable<? super E> key = (Comparable<? super E>) x;
        //k大于0时,循环遍历k
        while (k > 0) {
        	//获取k的父节点
            int parent = (k - 1) >>> 1;
            //获取父节点的元素e
            Object e = queue[parent];
            //如果key的值大于等于其父节点的元素,那么不需要进行调整,结束操作
            if (key.compareTo((E) e) >= 0)
                break;
            //否则的话将两个值进行对换,同时将k赋值为其父节点的位置
            queue[k] = e;
            k = parent;
        }
        queue[k] = key;
    }

    /**
     * 如果有指定的比较器,插入元素之后的调整方法
     * @param k
     * @param x
     */
    private void siftUpUsingComparator(int k, E x) {
        while (k > 0) {
            int parent = (k - 1) >>> 1;
            Object e = queue[parent];
            if (comparator.compare(x, (E) e) >= 0)
                break;
            queue[k] = e;
            k = parent;
        }
        queue[k] = x;
    }
    下面是一个使用自然排序的优先级队列中插入元素的实现过程:

java queue-PriorityQueue源码解析

java queue-PriorityQueue源码解析

java queue-PriorityQueue源码解析

4.PriorityQueue中remove(Object obj)方法的实现

    在优先级队列中删除一个指定的元素其操作过程同样分为几个步骤进行实现:

    (1)找到要删除的元素在数组中的索引位置,如果在数组中不存在,那么结束删除,返回false,其查找索引位置的方法是对数组进行遍历;

    (2)获取数组中最后一个元素的位置,也就是队列中的最后一个元素,如果要删除的元素就是最后一个元素,那么直接将最后一个位置的元素值置为null,结束操作;

    (3)获取最后一个元素的元素值,调用siftDown方法,进行向下调整,其主要的实现逻辑是将被删除元素的孩子结点中的最小值与替换的元素进行比较,进行上移或下移的操作,直到将原最后一个元素放到适合的位置,使其满足优先级队列的特性

    (4)如果原最后一个元素执行了上移操作,则返回这个元素,否则返回null

/**
     * 删除某一个元素
     */
    public boolean remove(Object o) {
    	//首先回去元素在数组中的索引位置
        int i = indexOf(o);
        if (i == -1)
            return false;
        else {
            removeAt(i);
            return true;
        }
    }
    
    /**
     * 获取某个元素的索引位置,对数组进行遍历
     * @param o
     * @return
     */
    private int indexOf(Object o) {
        if (o != null) {
            for (int i = 0; i < size; i++)
                if (o.equals(queue[i]))
                    return i;
        }
        return -1;
    }
    
    /**
     * 删除某个位置的元素
     * @param i
     * @return
     */
    private E removeAt(int i) {
        assert i >= 0 && i < size;
        modCount++;
        //s为数组中的最后一个元素的位置
        int s = --size;
        //如果删除的是最后一个元素,将最后一个位置的元素置为null
        if (s == i) // removed last element
            queue[i] = null;
        else {
        	//获取最后一个元素值
            E moved = (E) queue[s];
            //将最后一个位置的元素置为null
            queue[s] = null;
            siftDown(i, moved);
            //如果是最后一个元素替换到了删除的元素的位置,那么需要将元素进行向上比较,使其满足优先级队列的特性
            if (queue[i] == moved) {
                siftUp(i, moved);
                if (queue[i] != moved)
                    return moved;
            }
        }
        return null;
    }
    
    /**
     * 向下调整队列中的元素
     * @param k
     * @param x
     */
    private void siftDown(int k, E x) {
        if (comparator != null)
            siftDownUsingComparator(k, x);
        else
            siftDownComparable(k, x);
    }

    /**
     * 当无比较器时的下移操作的实现
     * @param k
     * @param x
     */
    private void siftDownComparable(int k, E x) {
        Comparable<? super E> key = (Comparable<? super E>)x;
        //half为队列的中间一半位置
        int half = size >>> 1;        // loop while a non-leaf
        //循环调整k,k小于中间位置索引值时循环
        while (k < half) {
        	//获取k的左孩子索引值
            int child = (k << 1) + 1; // assume left child is least
            //获取左孩子的值
            Object c = queue[child];
            //获取k的右孩子索引值
            int right = child + 1;
            //如果存在右孩子,并且右孩子的值小于左孩子的值,那么将c值赋值为右孩子的值,也就是取左右孩子中值较小的那位
            if (right < size &&
                ((Comparable<? super E>) c).compareTo((E) queue[right]) > 0)
                c = queue[child = right];
            //如果key值小于等于孩子结点中的值,不需要调整,结束循环
            if (key.compareTo((E) c) <= 0)
                break;
            //如果key值大于孩子结点中的最小值,那么将两个
            queue[k] = c;
            k = child;
        }
        queue[k] = key;
    }

    private void siftDownUsingComparator(int k, E x) {
        int half = size >>> 1;
        while (k < half) {
            int child = (k << 1) + 1;
            Object c = queue[child];
            int right = child + 1;
            if (right < size &&
                comparator.compare((E) c, (E) queue[right]) > 0)
                c = queue[child = right];
            if (comparator.compare(x, (E) c) <= 0)
                break;
            queue[k] = c;
            k = child;
        }
        queue[k] = x;
    }

下面是删除一个队列中的元素4的执行过程:

java queue-PriorityQueue源码解析

三、PriorityQueue总结

    上面主要讲了PriorityQueue的插入和删除元素的方法,对于该队列的其他方法,基本上都是基于这两个方法来实现的,在这里也就不一一讲解了,最后在总结一下优先级队列的特点:

    (1)普通队列都是先进先出的,但是优先级队列是根据优先级进行确定的,优先级最高的先出队列;

    (2)优先级队列类似于一棵完全二叉树,其内部实现使用的是数组;

    (3)优先级队列在数组中其元素不一定是完全有序的,但是在出队列时,其元素是有序的;

    (4)优先级队列插入和删除元素时,都需要对队列中的元素进行调整,其中remove()和add()方法的时间复杂度为O(log n),而remove(Obejct obj)和contaions()方法需要O(n)时间复杂度,取对头元素只需要O(1)时间;

    (5)优先级队列是非同步的,队列中不允许使用null元素。