Java：List（五）——PriorityQueue(优先级队列)

说明

PriorityQueue是优先级队列，它与Queue的区别在于，它的出队优先级与元素优先级相关——对它调用remove()或poll()方法时，返回的总是优先级最高的元素。

要使用PriorityQueue，需要给元素定义“优先级”。放入其中的元素，必须实现Comparable接口，PriorityQueue会根据排序结果决定出队优先级。如果元素没有实现Comparable接口，正如之前的TreeMap和TreeSet，可以传入一个Comparator对象。

注意

比较时返回-1说明优先级比较高。

模块：java.util.PriorityQueue

用法：

Queue <String> q = new PriorityQueue<> ();

例子：

一个银行排队业务：对User进行排队，如果User的号码以“V”开头，那它的优先级最高；两个同为“V”的客户，则按照号码数排队出队。

import java.util.Comparator;
import java.util.Queue;
import java.util.PriorityQueue;

public class Main {
    public static void main(String[] args) {
        Queue <User> q = new PriorityQueue<>(new Comparator<User>(){
            public int compare(User u1,User u2){
                if(u1.number.charAt(0) == u2.number.charAt(0))
                    return u1.number.compareTo(u2.number);
                else{
                    if(u1.number.charAt(0)=='V')
                        return -1;//想要先输出的就返回-1
                    else
                        return 1;
                }
            }
        });
        q.offer(new User("Bob", "A1"));
        q.offer(new User("Alice", "A2"));
        q.offer(new User("Boss", "V1"));
        System.out.println(q.poll()); // Boss/V1
        System.out.println(q.poll()); // Bob/A1
        System.out.println(q.poll()); // Alice/A2
        System.out.println(q.poll()); // null,因为队列为空
    }
}
class User{
    public final String name;
    public final String number;

    public User(String name,String number){
        this.name=name;
        this.number=number;
    }
    public String toString(){
        return name+"/"+number;
    }
}

不过上文中的比较逻辑存在问题，会把A10排到A2前面，可以修改为：

if(u1.number.charAt(0) == u2.number.charAt(0))
　　if (u1.number.length()==u2.number.length())
   　　 return u1.number.compareTo(u2.number);
　　else if(u1.number.length()>u2.number.length())
  　　  return 1;
　　else
  　　  return -1;