JAVA比较器的写法和优先队列的使用
标签：ati ash 数据 amp code turn 筛选 队列 ng2
1333. Filter Restaurants by Vegan-Friendly, Price and Distance
这道题做的过程中，由于语法不熟悉，花费了大量时间，需要掌握comparator自定义写法。
本题可以
1.用HashMap来存过滤后的数据，然后用自定义的比较器，在list中对符合要求的数据排序。
2.用PriorityQueue，自定义优先队列的排序方式，将筛选后的数据存入优先队列，即可自动有序。
class Solution {
public List filterRestaurants(int[][] restaurants, int veganFriendly, int maxPrice, int maxDistance) {
Mapint[]> map = new HashMap();
List IDs = new ArrayList();
for(int[] id : restaurants){
if(id[2] >= veganFriendly && id[3] maxDistance){
map.put(id[0], id);
IDs.add(id[0]);
}
}
//第一种写法
/* Collections.sort(IDs, (id1, id2) -> {
int rating1 = map.get(id1)[1];
int rating2 = map.get(id2)[1];
if(rating1 == rating2) return id2-id1;
return rating2-rating1;
});
*/
//第二种写法
IDs.sort(new Comparator(){
@Override
public int compare(Integer o1, Integer o2){
int rating1 = map.get(o1)[1];
int rating2 = map.get(o2)[1];
if(rating1 == rating2) return o2-o1;
return rating2-rating1;
}
});
return IDs;
}
}
class Solution {
public List filterRestaurants(int[][] restaurants, int veganFriendly, int maxPrice, int maxDistance) {
PriorityQueueint[]> pq = new PriorityQueue((a,b) -> a[1] == b[1] ? b[0]-a[0]:b[1]-a[1]);
List res = new ArrayList();
for(int[] id : restaurants){
if(id[2] >= veganFriendly && id[3] maxDistance)
pq.add(id);
}
while(!pq.isEmpty())
res.add(pq.remove()[0]);
return res;
}
}
JAVA比较器的写法和优先队列的使用
标签：ati ash 数据 amp code turn 筛选 队列 ng2
原文地址：https://www.cnblogs.com/yawenw/p/13021519.html
评论