温馨提示×

Java中priorityqueue自定义排序怎么实现

小亿
115
2023-11-22 01:37:07
栏目: 编程语言

在Java中,可以通过实现Comparator接口,并在PriorityQueue的构造函数中传入该Comparator对象来实现自定义排序。

以下是一个示例代码,按照字符串长度进行排序:

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

public class Main {
    public static void main(String[] args) {
        PriorityQueue<String> queue = new PriorityQueue<>(new StringLengthComparator());
        queue.add("Java");
        queue.add("Python");
        queue.add("C++");
        queue.add("Ruby");

        while (!queue.isEmpty()) {
            System.out.println(queue.poll());
        }
    }

    static class StringLengthComparator implements Comparator<String> {
        @Override
        public int compare(String s1, String s2) {
            return Integer.compare(s1.length(), s2.length());
        }
    }
}

输出结果:

C++
Java
Ruby
Python

在这个示例中,我们创建了一个PriorityQueue对象,传入了StringLengthComparator对象。StringLengthComparator实现了Comparator接口,并重写了compare方法来定义自定义的排序逻辑,比较字符串的长度。

在PriorityQueue中,每次从队列中取出元素时,会根据Comparator的compare方法进行比较并排序。因此,输出结果会按照字符串的长度从小到大排序。

0