温馨提示×

priorityqueue java怎样自定义比较器

小樊
81
2024-11-16 02:55:20
栏目: 编程语言

在Java中,要自定义PriorityQueue的比较器,您需要创建一个实现Comparator接口的类,并重写compare方法。然后,您可以将这个自定义比较器传递给PriorityQueue的构造函数。以下是一个示例:

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

// 自定义比较器类
class CustomComparator implements Comparator<Integer> {
    @Override
    public int compare(Integer o1, Integer o2) {
        // 在这里实现自定义的比较逻辑
        // 例如,这里我们按照降序排列
        return o2 - o1;
    }
}

public class Main {
    public static void main(String[] args) {
        // 创建一个整数类型的PriorityQueue,并使用自定义比较器
        PriorityQueue<Integer> priorityQueue = new PriorityQueue<>(new CustomComparator());

        // 向队列中添加元素
        priorityQueue.add(3);
        priorityQueue.add(1);
        priorityQueue.add(2);

        // 输出队列中的元素
        while (!priorityQueue.isEmpty()) {
            System.out.println(priorityQueue.poll());
        }
    }
}

在这个示例中,我们创建了一个名为CustomComparator的自定义比较器类,它实现了Comparator<Integer>接口。在compare方法中,我们实现了自定义的比较逻辑(这里是按降序排列)。然后,我们创建了一个整数类型的PriorityQueue,并将自定义比较器传递给它。最后,我们向队列中添加了一些元素,并输出了它们。

0