JAVA比较器的写法和优先队列的使用
1333. Filter Restaurants by Vegan-Friendly, Price and Distance
这道题做的过程中,由于语法不熟悉,花费了大量时间,需要掌握comparator自定义写法。
本题可以
1.用HashMap来存过滤后的数据,然后用自定义的比较器,在list中对符合要求的数据排序。
2.用PriorityQueue,自定义优先队列的排序方式,将筛选后的数据存入优先队列,即可自动有序。
class Solution { public List<Integer> filterRestaurants(int[][] restaurants, int veganFriendly, int maxPrice, int maxDistance) { Map<Integer, int[]> map = new HashMap<>(); List<Integer> IDs = new ArrayList<>(); for(int[] id : restaurants){ if(id[2] >= veganFriendly && id[3] <= maxPrice && id[4] <= 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<Integer>(){ @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<Integer> filterRestaurants(int[][] restaurants, int veganFriendly, int maxPrice, int maxDistance) { //PriorityQueue<int[]> pq = new PriorityQueue<>((a,b) -> b[1] == a[1]? b[0] - a[0]: b[1]-a[1]) PriorityQueue<int[]> pq = new PriorityQueue<>(new Comparator<int[]>(){ @Override public int compare(int[] o1, int[] o2){ if(o1[1] == o2[1]) return o2[0]-o1[0]; return o2[1]-o1[1]; } }); List<Integer> res = new ArrayList<>(); for(int[] id : restaurants){ if(id[2] >= veganFriendly && id[3] <= maxPrice && id[4] <= maxDistance) pq.add(id); } while(!pq.isEmpty()) res.add(pq.remove()[0]); return res; } }