1333. 餐厅过滤器

原题戳这里

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 <= id, rating, price, distance <= 10^5
1 <= maxPrice, maxDistance <= 10^5
veganFriendly 和 veganFriendly 的值为 0 或 1 。
所有 id 各不相同。

2. 分析

2.1 筛选

题目并不算难，我们要做的就是遍历并找出满足这些条件的餐厅

我们遍历restaurants这个二维数组并且每个遍历到的一维数组设为restaurant

(veganFriendly == 0 || (veganFriendly + restaurant[2] == 2))
&& restaurant[3] <= maxPrice 
&& restaurant[4] <= maxDistance

veganFriendly == 0 || (veganFriendly + restaurant[2] == 2)

如果不需要素食友好，即veganFriendly == 0 或者 veganFriendly 和餐厅的素食友好值(即restaurant[2])都为1，则素食友好这个方面满足条件
restaurant[3] <= maxPrice

餐厅的最大价格要小于参数给出的最大价格。

restaurant[4] <= maxDistance

餐厅到你的距离应该小于参数所给的最大距离

2.2 排序

我们把满足如上条件的一维数组都存起来，我用的是ArrayList<int[]>来存的，所以上一步是比较简单的；困难的地方主要是排序。这里我们调用ArrayList的sort方法，这个方法需要我们传入一个Comparator匿名类。首先我们根据rating（即restaurant[1]）来排序，如果rating相同，那么我们就使用id（即restaurant[0]）来排序。那么我们的Comparator可以写成

new Comparator<int[]>() {
   
    @Override
    public int compare(int[] o1, int[] o2) {
   
        if (o1[1] == o2[1]) {
   
            return o2[0] - o1[0];
        } else {
   
            return o2[1] - o1[1];
        }
    }
}

使用lamda表达式的话就是

(o1, o2) -> {
   
    if (o1[1] == o2[1]) {
   
        return o2[0] - o1[0];
    } else {
   
        return o2[1] - o1[1];
    }
}

3. 代码

public List<Integer> filterRestaurants(int[][] restaurants, int veganFriendly,
                                       int maxPrice, int maxDistance) {
   
    List<int[]> resList = new ArrayList<>();
    //筛选符合条件的数组
    for (int[] restaurant : restaurants) {
   
        if ((veganFriendly == 0 || (veganFriendly + restaurant[2] == 2))
            && restaurant[3] <= maxPrice && restaurant[4] <= maxDistance) {
   
            resList.add(restaurant);
        }
    }

    //排序
    resList.sort((o1, o2) -> {
   
        if (o1[1] == o2[1]) {
   
            return o2[0] - o1[0];
        } else {
   
            return o2[1] - o1[1];
        }
    });

    List<Integer> res = new ArrayList<>();
    //遍历获得id
    for (int[] i :
         resList) {
   
        res.add(i[0]);
    }
    return res;
}

（8月20日成绩）