Java中的 PriorityQueue 比较器() 方法

简介

Java中的 PriorityQueue是一种优先队列，它基于堆数据结构实现。它的元素按照优先级的顺序被移除，因此队列中的元素不是按照它们被加入的顺序排列的。

在PriorityQueue中，我们可以自定义比较器（Comparator）来自定义元素的优先级。在这篇文章中，我们将学习如何使用比较器来控制PriorityQueue的行为。

Comparator接口

Comparator是一个函数式接口，它定义了一个函数来比较两个对象。这个接口只有一个方法，就是compare()方法。该方法接收两个参数，一个是第一个元素，另一个是第二个元素，它们将被比较。该方法返回一个整数，表示它们之间的比较结果。如果返回值小于零，则表示第一个元素优先级较高；如果返回值大于零，则表示第二个元素优先级较高；如果返回值等于零，则表示它们的优先级相等。

使用Comparator来自定义优先级

我们可以使用Comparator接口来自定义PriorityQueue的元素优先级。我们可以创建一个新的类，实现Comparator接口，并实现compare()方法。在这个方法中，我们可以定义我们自己的比较逻辑。

以下是一个示例，它演示了如何使用Comparator接口来比较字符串，并按字母顺序对它们进行排序。请注意，我们使用了PriorityQueue构造函数的重载来指定比较器。

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

public class PriorityQueueComparatorExample {

    public static void main(String[] args) {

        // Create a new priority queue and specify a Comparator
        PriorityQueue<String> pq = new PriorityQueue<>(new Comparator<String>() {
            @Override
            public int compare(String s1, String s2) {
                return s1.compareTo(s2);
            }
        });

        // Add some strings to the queue
        pq.add("banana");
        pq.add("apple");
        pq.add("pear");

        // Remove and print each string from the queue
        while (!pq.isEmpty()) {
            System.out.println(pq.remove());
        }
    }
}

输出结果：

apple
banana
pear

lambda表达式

Java 8引入了lambda表达式，它使得编写简短的比较器变得更加容易。下面是上面的示例，使用lambda表达式来定义比较器：

// Create a new priority queue and specify a Comparator using a lambda expression
PriorityQueue<String> pq = new PriorityQueue<>((s1, s2) -> s1.compareTo(s2));

结论

在本文中，我们学习了如何使用Comparator接口来定义PriorityQueue元素之间的比较逻辑。我们演示了如何使用匿名内部类和lambda表达式来创建比较器。现在，您可以控制PriorityQueue的元素插入/删除顺序并应用它们的优先级，以满足您的需求。