我有一个包含以下字段的实体类:id、orderNo。每个实体都必须存储在一个 java 优先级队列中。id 在 1 - 3000 之间的元素具有更高的优先级,必须以 orderNo 的升序存储在 id > 3000 的元素之上。 ids > 3000 的元素以 orderNo 的升序存储在更高优先级的元素(ids 1 - 3000)。
例如:
(1st insertion to queue: id=4000 orderNo=1)
(2nd insertion to queue: id=5000 orderNo=2)
(3rd insertion to queue: id=100 orderNo=3)
(4th insertion to queue: id=50 orderNo=4)
预期排序顺序:
(id=100 orderNo=3)
(id=50 orderNo=4)
(id=4000 orderNo=1)
(id=5000 orderNo=2)
订单实体类:
public class OrderEntity implements Comparable<OrderEntity> {
private int id;
private int getOrderNo;
public int getId() {
return id;
}
public void setId(int id) {
this.id = id;
}
public int getOrderNo() {
return getOrderNo;
}
public void setOrderNo(int getOrderNo) {
this.getOrderNo = getOrderNo;
}
@Override
public int compareTo(OrderEntity arg0) {
if ((this.getId() >= 1 && this.getId() <= 3000) && (arg0.getId() >= 1 && arg0.getId() <= 3000)) {
if (this.getOrderNo() > arg0.getOrderNo()) {
return 1;
} else {
return 0;
}
} else if ((this.getId() <= 3000) && (arg0.getId() > 3000)) {
return 1;
} else if ((this.getId() > 3000) && (arg0.getId() <= 3000)) {
return 1;
} else if ((this.getId() > 3000) && (arg0.getId() > 3000)) {
if (this.getOrderNo() > arg0.getOrderNo()) {
return 1;
} else {
return 0;
}
} else {
return 0;
}
}
}
订单处理器类:
public class OrderProcessor {
private static int count;
static Queue<OrderEntity> pq = new PriorityQueue<>();
public String createOrder(int id) {
OrderEntity orderEntity = new OrderEntity();
orderEntity.setId(id);
count = count + 1;
orderEntity.setOrderNo(count);
pq.add(orderEntity);
String res = "";
for (OrderEntity rd : pq) {
res = res + rd.getId() + " " + rd.getOrderNo() + "\n";
}
return res.trim();
}
}
在这种情况下,对象的自然顺序与您的特殊要求不同,最好不要使用,Comparable
因为它将来可能有其他用途。因此,剩下的解决方案正在使用Comparator
它非常适合您的问题,因为您的OrderEntity
班级不会依赖于这个特殊的比较。以下是显示解决方案的示例代码:
import java.util.Comparator;
import java.util.PriorityQueue;
public class OrderProcessor {
public static void main(String[] args) {
PriorityQueue<OrderEntity> q = new PriorityQueue<>(new OrderEntityComparator());
q.add(new OrderEntity(4000, 1));
q.add(new OrderEntity(5000, 2));
q.add(new OrderEntity(100, 3));
q.add(new OrderEntity(50, 4));
while(!q.isEmpty())
System.out.println(q.poll());
}
public static class OrderEntityComparator implements Comparator<OrderEntity> {
@Override
public int compare(OrderEntity o1, OrderEntity o2) {
if(o1.getId() <= 3000 && o2.getId() <= 3000)
return Integer.compare(o1.getOrderNo(), o2.getOrderNo());
if(o1.getId() > 3000 && o2.getId() > 3000)
return Integer.compare(o1.getOrderNo(), o2.getOrderNo());
if(o1.getId() <= 3000 && o2.getId() > 3000)
return -1;
return 1;
}
}
public static class OrderEntity {
private int id;
private int orderNo;
public OrderEntity(int id, int orderNo) {
this.id = id;
this.orderNo = orderNo;
}
public int getId() {
return id;
}
public void setId(int id) {
this.id = id;
}
public int getOrderNo() {
return orderNo;
}
public void setOrderNo(int orderNo) {
this.orderNo = orderNo;
}
@Override
public String toString() {
return "OrderEntity{" +
"id=" + id +
", orderNo=" + orderNo +
'}';
}
}
}
编辑:
如果您不想通过调用poll
方法删除元素,则必须对数组或列表中的元素进行排序,如下所示:
OrderEntity[] a = new OrderEntity[q.size()];
q.toArray(a);
Arrays.sort(a, new OrderEntityComparator());
for(OrderEntity entity : a)
System.out.println(entity);
事实上,在这种情况下,您不需要使用 aPriorityQueue
并且对 List 或数组进行简单排序即可完成这项工作。
本文收集自互联网,转载请注明来源。
如有侵权,请联系[email protected] 删除。
我来说两句