在Java中,用优先队列对类进行自定义排序
时间: 2024-05-05 07:20:16 浏览: 89
优先队列-java可以选择属性和升序降序
要在Java中使用优先队列对类进行自定义排序,需要定义一个实现了Comparable接口的类。该接口定义了一个compareTo()方法,该方法接受一个对象作为参数,并返回一个整数值,表示当前对象与参数对象的比较结果。
比较结果应该满足以下规则:
- 返回负整数表示当前对象小于参数对象;
- 返回正整数表示当前对象大于参数对象;
- 返回零表示当前对象等于参数对象。
例如,假设我们要创建一个Person类,它有两个属性:姓名和年龄。我们想按年龄从小到大对Person对象进行排序,代码如下:
```
public class Person implements Comparable<Person> {
private String name;
private int age;
public Person(String name, int age) {
this.name = name;
this.age = age;
}
public String getName() {
return name;
}
public int getAge() {
return age;
}
@Override
public int compareTo(Person o) {
return Integer.compare(this.age, o.age);
}
}
```
在这个类中,我们实现了Comparable接口,并重写了compareTo()方法。该方法比较当前对象的年龄属性和参数对象的年龄属性,并返回比较结果。我们使用Integer.compare()方法来比较两个年龄属性,这是一个非常方便和安全的方法。
接下来,我们可以使用PriorityQueue类来创建优先队列,并将Person对象添加到队列中。PriorityQueue类会自动按照compareTo()方法的结果对对象进行排序。
例如,我们可以创建一个按年龄排序的Person对象队列:
```
PriorityQueue<Person> queue = new PriorityQueue<>();
queue.add(new Person("Tom", 25));
queue.add(new Person("Mary", 30));
queue.add(new Person("John", 20));
```
在这个例子中,我们使用add()方法将三个Person对象添加到队列中。由于我们已经实现了Comparable接口并重写了compareTo()方法,队列会自动按照年龄从小到大对Person对象进行排序。
我们可以使用poll()方法从队列中取出对象,并检查它们的顺序:
```
while (!queue.isEmpty()) {
Person p = queue.poll();
System.out.println(p.getName() + " " + p.getAge());
}
```
这将输出以下内容:
```
John 20
Tom 25
Mary 30
```
我们可以看到,Person对象按照年龄从小到大排序,正是我们期望的结果。
阅读全文