Java中的PriorityQueue的使用提示

起源

在Java中使用PriorityQueue的主要目的是使用堆这个数据结构，为了解决第K大/小的问题。。

深入理解

基本上自己定义的问题，需要重写比较器(Comparator),但是在重写比较器之前，需要知道初始比较器是怎么个作用。
这里以PriorityQueue Integer为例说明，默认是一个小堆，每次poll()出来的都第K大的数的最小者，但是要设置第k大，还要用if语句过滤，在里面poll()出元素，下面以具体例子说明。

实例

import java.util.PriorityQueue;
import java.util.Random;
import java.util.Comparator;


public class PriorityQueueTutorial {

    public static void main(String[] args) {
        // TODO 自动生成的方法存根
          PriorityQueue<Student> queue = new PriorityQueue<Student>(5,
                    new Comparator<Student>() {
                      public int compare(Student s1, Student s2) {
                        //选取成绩最好的人
                        return s1.grade - s2.grade;
                      }
                    });

            for (int i = 1; i <= 100; i++) {
                queue.add(new Student("s" + i, (new Random().nextInt(1000))));
                if (queue.size() > 5){
                    queue.poll();
                }
            }
            while (!queue.isEmpty()) {
                  System.out.println(queue.poll().toString());
            }

    }

}

class Student {
    String name;
    int grade;
    public Student(String name, int grade)
    {
        this.name = name;
        this.grade = grade;
    }

    public String toString() {
        return name + " " + grade;
    }
}

上面的例子是选出5个成绩最好的学生成绩。也是默认比较器。

那么如果要找5个成绩最差的学生呢？？
改下比较器：
return s2.grade - s1.grade;

举一反三

其他问题比如距离最近，距离最远就举一反三就可以