对于Java中priorityQueue的理解

定义

priorityQueue是Java中的一种无界优先队列，具体实现是二叉小顶堆，是一颗完全二叉树，实现了Queue接口，不允许null元素插入，初始时是一个Object[]，随着元素的插入可自动扩容，入队后元素按自然顺序从小到大排列，也可以重构comparator方法来排列

优先队列

保证每次取出的元素都是队列中权值最小那个

小顶堆

任意一个非叶子结点的权值，都不大于其左右子结点的权值

具体实现

priorityQueue具体实现是数组，实现小顶堆后，其父结点和子结点满足以下关系

• leftNode = ParentNode * 2 + 1
• rightNode = ParentNode * 2 + 2
• parentNode = (currentNode - 1) / 2

方法与时间复杂度

方法	时间复杂度
peek()	O(1)
element()	O(1)
add()	O(log(N))
offer()	O(log(N)
remove()	O(log(N))
poll()	O(log(N))

方法的区别

• add(E e)和offer(E e):都是向队列中插入元素，不同之处是插入失败时前者抛出异常，后者返回false

//offer(E e)
public boolean offer(E e) {
    if (e == null)//不允许放入null元素
        throw new NullPointerException();
    modCount++;
    int i = size;
    if (i >= queue.length)
        grow(i + 1);//自动扩容
    size = i + 1;
    if (i == 0)//队列原来为空，这是插入的第一个元素
        queue[0] = e;
    else
        siftUp(i, e);//调整
    return true;
}
/*
	1、其中，grow()是让队列扩容，就是申请一个更大的数组
	2、siftUp(int i, E e)用于插入元素e之后调整堆结构以保持堆的特性
*/
//siftUp()
private void siftUp(int k, E x) {
    while (k > 0) {
        int parent = (k - 1) >>> 1;//parentNo = (nodeNo-1)/2
        Object e = queue[parent];
        if (comparator.compare(x, (E) e) >= 0)//调用比较器的比较方法
            break;
        queue[k] = e;
        k = parent;
    }
    queue[k] = x;
}
/*
	从k指定的位置开始，将x逐层与当前点的parent进行比较并
	交换，直到满足x >= queue[parent]为止
*/

• element()和peek():都是获取但不删除队首元素，区别是获取失败时前者抛出异常，后者返回null；根据堆的性质，堆顶是整个堆中权值最小的元素，按照数组的小标就是第0个元素，所以可以直接返回下标为0的元素即可

//peek()
public E peek() {
    if (size == 0)
        return null;
    return (E) queue[0];//0下标处的那个元素就是最小的那个
}

• remove()和poll():都是获取并删除队首元素，区别是获取失败时前者抛出异常，后者返回null

//poll()
public E poll() {
    if (size == 0)
        return null;
    int s = --size;
    modCount++;
    E result = (E) queue[0];//0下标处的那个元素就是最小的那个
    E x = (E) queue[s];
    queue[s] = null;
    if (s != 0)
        siftDown(0, x);//调整
    return result;
}
//由于删除元素会破坏堆结构，所以需要siftDown()来调整
//siftDown()
private void siftDown(int k, E x) {
    int half = size >>> 1;
    while (k < half) {
        //首先找到左右孩子中较小的那个，记录到c里，并用child记录其下标
        int child = (k << 1) + 1;//leftNo = parentNo*2+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;//然后用c取代原来的值
        k = child;
    }
    queue[k] = x;
}
/*
	从k指定的位置开始，将x逐层向下与当前点的左右孩子中较小的
	那个交换，直到x小于或等于左右孩子中的任何一个为止
*/

• remove(Object o):用于删除队列中和o相等的元素，删除会出现两种情况：①删除的是最后一个元素，直接删除，不需要调整堆结构；②删除的是不是最后一个元素，则从删除的那个元素开始，以最后一个元素为参照物调用一次siftDown()来调整堆结构

//remove(Object o)
public boolean remove(Object o) {
    //通过遍历数组的方式找到第一个满足o.equals(queue[i])元素的下标
    int i = indexOf(o);
    if (i == -1)
        return false;
    int s = --size;
    if (s == i) //情况1
        queue[i] = null;
    else {
        E moved = (E) queue[s];
        queue[s] = null;
        siftDown(i, moved);//情况2
        ......
    }
    return true;
}

实例

这里我写了一个给无序的标签按其权值大小排序的封装

HotTagCache类：

public class HotTagCache {
    private List<String> hots = new ArrayList<>();

    /**
     * 给无序的标签元素按其所占的权重值从大到小排序
     *
     * @param tags 无序的标签<标签名，权值>
     */
    public void sortTags(Map<String, Integer> tags) {
        int max = 10;
        //新建一个最大长度为10的优先级队列
        PriorityQueue<HotTagDTO> priorityQueue = new PriorityQueue<>(max);

        tags.forEach((name, priority) -> {
            HotTagDTO hotTagDTO = new HotTagDTO();
            hotTagDTO.setName(name);
            hotTagDTO.setPriority(priority);
            if (priorityQueue.size() < max) {
                //如果当前队列长度小于最大长度则直接入队
                priorityQueue.add(hotTagDTO);
            } else {
                //队列已满，取出队首元素作比较，大于队首元素则入队
                HotTagDTO minHot = priorityQueue.peek();
                if (hotTagDTO.compareTo(minHot) > 0) {
                    priorityQueue.poll();
                    priorityQueue.add(hotTagDTO);
                }
            }
        });
        List<String> sortedTags = new ArrayList<>();
        //依次取出队列元素并add进入list里
        HotTagDTO poll = priorityQueue.poll();
        while (poll != null) {
            //List里的add(index, Element)：在索引位置采用头插法插入元素
            sortedTags.add(0, poll.getName());
            poll = priorityQueue.poll();
        }

        hots = sortedTags;
    }

}

HotTagDTO类，实现Comparable接口，重载compareTo()构造排序规则：

public class HotTagDTO implements Comparable {
    private String name;
    private Integer priority;

    /**
     * 重载compareTo方法，自己构造排序规则
     *
     * @param o
     * @return
     */
    @Override
    public int compareTo(Object o) {
        return this.getPriority() - ((HotTagDTO) o).getPriority();
    }
}

参考

• JAVA中PRIORITYQUEUE详解