带自定义比较器的Java PriorityQueue [英] Java PriorityQueue with custom Comparator

问题描述

我使用的是PriorityQueue和我自己的比较器，但不知何故，最终结果并不总是好的。 我应该按照平均成绩，而不是名字，而不是ID来排序。不。最后，它应该返回已排序的队列中剩余的名称。剩下的名字是好的，但他们的顺序不是。 输入(名称，等级平均值，id.no)：

add John 3,75 50
add Mark 3,8 24
add Shafaet 3,7 35
poll
poll
add Samiha 3,85 36
poll
add Ashley 3,9 42
add Maria 3,6 46
add Anik 3,95 49
add Dan 3,95 50
poll

预期输出：

Dan
Ashley
Shafaet
Maria

我的结果：

Dan
Ashley
Maria
Shafaet

你能帮我找出这个问题吗？ 提前感谢您！

class StComp implements Comparator<Students> {
        @Override
        public int compare(Students st1, Students st2) {
            if (st1.getCgpa() == st2.getCgpa()) {
                if (st1.getName().equals(st2.getName()))
                    return st1.getId() - st2.getId();
                else
                    return st1.getName().compareTo(st2.getName());
            }
            else
                return (st1.getCgpa() < st2.getCgpa()) ? 1 : -1;
        }
    }

    StComp stComp = new StComp();
    PriorityQueue<Students> pq = new PriorityQueue<Students>(2, stComp);

推荐答案

您的Comparator正确。问题是您最有可能使用它的Iterator遍历列表。PriorityQueue documentation状态：

方法迭代器()中提供的迭代器不能保证 以任何特定顺序遍历优先级队列的元素。

如果像这样迭代PriorityQueue，您应该会看到正确的结果：

while (!pq.isEmpty())
    System.out.println(pq.poll().getName());
}

我已在此答案的末尾添加了一个示例以充分演示。

---

如果您不想清除PriorityQueue，您可以执行几项操作。就我个人而言，我不推荐这两种方法中的任何一种，因为最初选择的PriorityQueue对于用例是不正确的，因为它们不打算被迭代。

您可以将PriorityQueue复制到一个数组中，使用您的Comparator实现对它们进行排序，然后迭代已排序的数组，例如：

Student[] students = pq.toArray(new Student[pq.size()]);
Arrays.sort(students, new StComp());
for (Student s : students) {
    System.out.println(s.getName() + " " + s.getCgpa() + " " + s.getId());
}

或在轮询时将它们添加到某种Collection，然后再将它们添加回PriorityQueue，例如：

Collection<Student> temp = new LinkedList<>();
while (!pq.isEmpty()) {
    Student s = pq.poll();
    System.out.println(s.getName() + " " + s.getCgpa() + " " + s.getId());
    temp.add(s);
}
pq.addAll(temp);

---

使用您的数据演示的示例：

Main

public class Main {

    public static void main(String[] args) {
        PriorityQueue<Student> pq = new PriorityQueue<>(new StComp());
        pq.add(new Student("John", 75, 50)); // Student name, grade average, id
        pq.add(new Student("Mark", 8, 24));
        pq.add(new Student("Shafaet", 7, 35));
        pq.poll();
        pq.poll();
        pq.add(new Student("Samiha", 85, 36));
        pq.poll();
        pq.add(new Student("Ashley", 9, 42));
        pq.add(new Student("Maria", 6, 46));
        pq.add(new Student("Anik", 95, 49));
        pq.add(new Student("Dan", 95, 50));
        pq.poll();

        // Not guaranteed to be in priorty order
        System.out.println("Using PriorityQueue's Iterator, may not be in the correct priority order.");
        for (Student s : pq) {
            System.out.println(s.getName() + " " + s.getCgpa() + " " + s.getId());
        }

        // Correct order, but removes from the Priority Queue
        System.out.println("
Iterating until empty using PriorityQueue.poll(), will be in the correct order.");
        while (!pq.isEmpty()) {
            Student s = pq.poll();
            System.out.println(s.getName() + " " + s.getCgpa() + " " + s.getId());
        }
    }

}

学生(已重命名，应为单数)

public class Student {

    private double cgpa;
    private String name;
    private int id;

    public Student(String name, double cgpa, int id) {
        this.name = name;
        this.cgpa = cgpa;
        this.id = id;
    }

    public String getName() {
        return name;
    }

    public int getId() {
        return id;
    }

    public double getCgpa() {
        return cgpa;
    }

}

StComp(问题逻辑不变)

public class StComp implements Comparator<Student> {

    @Override
    public int compare(Student st1, Student st2) {
        if (st1.getCgpa() == st2.getCgpa()) {
            if (st1.getName().equals(st2.getName())) {
                return st1.getId() - st2.getId();
            } else {
                return st1.getName().compareTo(st2.getName());
            }
        } else {
            return (st1.getCgpa() < st2.getCgpa()) ? 1 : -1;
        }
    }
}

输出(至少对我来说，第一个Iterator变量的结果可能有所不同)

Using PriorityQueue's Iterator, may not be in the correct priority order.
Dan 95.0 50
Ashley 9.0 42
Maria 6.0 46
Shafaet 7.0 35

Iterating until empty using PriorityQueue.poll(), will be in the correct order.
Dan 95.0 50
Ashley 9.0 42
Shafaet 7.0 35
Maria 6.0 46

这篇关于带自定义比较器的Java PriorityQueue的文章就介绍到这了，希望我们推荐的答案对大家有所帮助，也希望大家多多支持IT屋！