1. 问题描述:
给你一个餐馆信息数组 restaurants,其中 restaurants[i] = [idi, ratingi, veganFriendlyi, pricei, distancei]。你必须使用以下三个过滤器来过滤这些餐馆信息。
其中素食者友好过滤器 veganFriendly 的值可以为 true 或者 false,如果为 true 就意味着你应该只包括 veganFriendlyi 为 true 的餐馆,为 false 则意味着可以包括任何餐馆。此外,我们还有最大价格 maxPrice 和最大距离 maxDistance 两个过滤器,它们分别考虑餐厅的价格因素和距离因素的最大值。
过滤后返回餐馆的 id,按照 rating 从高到低排序。如果 rating 相同,那么按 id 从高到低排序。简单起见, veganFriendlyi 和 veganFriendly 为 true 时取值为 1,为 false 时,取值为 0 。
示例 1:
输入:restaurants = [[1,4,1,40,10],[2,8,0,50,5],[3,8,1,30,4],[4,10,0,10,3],[5,1,1,15,1]], veganFriendly = 1, maxPrice = 50, maxDistance = 10
输出:[3,1,5]
解释:
这些餐馆为:
餐馆 1 [id=1, rating=4, veganFriendly=1, price=40, distance=10]
餐馆 2 [id=2, rating=8, veganFriendly=0, price=50, distance=5]
餐馆 3 [id=3, rating=8, veganFriendly=1, price=30, distance=4]
餐馆 4 [id=4, rating=10, veganFriendly=0, price=10, distance=3]
餐馆 5 [id=5, rating=1, veganFriendly=1, price=15, distance=1]
在按照 veganFriendly = 1, maxPrice = 50 和 maxDistance = 10 进行过滤后,我们得到了餐馆 3, 餐馆 1 和 餐馆 5(按评分从高到低排序)。
示例 2:
输入:restaurants = [[1,4,1,40,10],[2,8,0,50,5],[3,8,1,30,4],[4,10,0,10,3],[5,1,1,15,1]], veganFriendly = 0, maxPrice = 50, maxDistance = 10
输出:[4,3,2,1,5]
解释:餐馆与示例 1 相同,但在 veganFriendly = 0 的过滤条件下,应该考虑所有餐馆。
示例 3:
输入:restaurants = [[1,4,1,40,10],[2,8,0,50,5],[3,8,1,30,4],[4,10,0,10,3],[5,1,1,15,1]], veganFriendly = 0, maxPrice = 30, maxDistance = 3
输出:[4,5]
提示:
1 <= restaurants.length <= 10^4
restaurants[i].length == 5
1 <= idi, ratingi, pricei, distancei <= 10^5
1 <= maxPrice, maxDistance <= 10^5
veganFriendlyi 和 veganFriendly 的值为 0 或 1 。
所有 idi 各不相同。
来源:力扣(LeetCode)
链接:https://leetcode-cn.com/problems/filter-restaurants-by-vegan-friendly-price-and-distance
2. 思路分析:
① 题目其实还是比较好理解的,我们的任务是需要选择出评分高的餐厅,评分相同的情况下选择id较高的餐厅,这个自然而然地想到使用map进行映射(map映射与计数的功能是非常好的),其实我们很多题目都是可以借助于map来实现的,我们可以将评分相同的餐厅是可以放在一起的,键为餐厅的评分,因为可能存在多个评分相同的餐厅对应的序号,所以值设置为List<Integr>来连接多个餐厅的序号
② 我们可以在for循环中首先过滤掉那些不满足条件的餐厅,在符合条件的情况下将评分相同的餐厅放在一起,因为我们最后要求的答案是评分高的排列在前面,所以我们可以使用TreeMap来实现map这个接口,这样可以自定义键的排序规则为从大到小进行排序,值为List<Integer>,最后我们在遍历的时候是按照评分高的餐厅进行遍历的,并且这里有一个坑是给出的餐厅的序号并不是按照从小到大的顺序进行排列的,所以我们在遍历List的时候需要先对List进行排序这样得到的结果才是正确的
3. 代码如下:
class Solution {
/*可以使用map来进行解决, 将相同评分的放在一个键中, 值对应的就是对应的id*/
/*这里有一个坑是给出的不一定按照餐厅序号从小到大来的*/
public List<Integer> filterRestaurants(int[][] restaurants, int veganFriendly, int maxPrice, int maxDistance) {
/*键自定义为从大到小进行排序*/
Map<Integer, List<Integer>> map = new TreeMap<>(new Comparator<Integer>() {
@Override
public int compare(Integer o1, Integer o2) {
return o2 - o1;
}
});
for (int []cur : restaurants){
if (veganFriendly == 1 && cur[2] == 0) continue;
if (maxPrice < cur[3]) continue;
if (maxDistance < cur[4]) continue;
List<Integer> t = map.getOrDefault(cur[1], new ArrayList<>());
t.add(cur[0]);
map.put(cur[1], t);
}
List<Integer> res = new ArrayList<>();
for (Map.Entry<Integer, List<Integer>> entry : map.entrySet()){
List<Integer> cur = entry.getValue();
Collections.sort(cur);
for (int i = cur.size() - 1; i >= 0; --i){
res.add(cur.get(i));
}
}
return res;
}
}