在写算法题的时候,经常会使用到优先队列PriorityQueue
,此时需要构建一个比较器Comparator
,如果能借助java的API快速的构建一个比较器实现优先队列,能够使自己的代码更加简练。下面是常见的比较器写法:
最常见的两个比较器写法
PriorityQueue<Integer> min = new PriorityQueue<>(Comparator.naturalOrder());
PriorityQueue<Integer> max = new PriorityQueue<>(Comparator.reverseOrder());
naturalOrder,为正序比较器,构建出来的堆为小跟堆
reverseOrder,为逆序,构建出来的堆为大根堆
对象比较
Comparator类中提供了几个已经写好的方法。用于正序比较
如进行Int数据比较的comparingInt
,比较普通类型的comparing
,在构建时只需要使用lambada表达式,即可快速完成构建
//PriorityQueue<Node> nodeq = new PriorityQueue<>((o1,o2)->(o1.value-o2.value));
PriorityQueue<Node> nodeq = new PriorityQueue<>(Comparator.comparingInt(o -> o.value));
//PriorityQueue<Node> nodeq2 = new PriorityQueue<>((o1, o2) -> o1.str.compareTo(o2.str));
PriorityQueue<Node> nodeq2 = new PriorityQueue<>(Comparator.comparing(o -> o.str));
如果想完成逆序比较的话,就只能借助最原始的lambada表达式来完成了
如果需求是完成更为复杂的比较(例如有三个字段,需要逐一进行比较),则可以借助thenXX方法:
Comparator<Node> c= Comparator.comparing((Node o) ->o.str).thenComparingInt((Node o)->o.value);
大量的借用java的API和Lambada表达式可以实现代码的简洁