排序题目：餐厅过滤器

题目

标题和出处

标题：餐厅过滤器

出处：1333. 餐厅过滤器

难度

3 级

题目描述

要求

给定一个餐馆信息数组 $\text{restaurants}$，其中 ${\text{restaurants[i] = [id}}_{\text{i}}{\text{, rating}}_{\text{i}}{\text{, veganFriendly}}_{\text{i}}{\text{, price}}_{\text{i}}{\text{, distance}}_{\text{i}}\text{]}$。需要使用三个过滤器来过滤这些餐馆信息。

素食者友好过滤器 $\text{veganFriendly}$ 的值可以为 $\text{true}$ 或者 $\text{false}$，如果为 $\text{true}$ 就意味着应该只包括 ${\text{veganFriendly}}_{\text{i}}$ 为 $\text{true}$ 的餐馆，为 $\text{false}$ 则意味着可以包括任何餐馆。此外，还有最大价格 $\text{maxPrice}$ 和最大距离 $\text{maxDistance}$ 两个过滤器，分别表示考虑餐厅的价格因素和距离因素的最大值。

过滤后返回餐馆的 $\text{id}$，按照 $\text{rating}$ 从高到低排序。如果 $\text{rating}$ 相同，那么按 $\text{id}$ 从高到低排序。简单起见，${\text{veganFriendly}}_{\text{i}}$ 和 $\text{veganFriendly}$ 为 $\text{true}$ 时取值为 $\text{1}$，为 $\text{false}$ 时取值为 $\text{0}$。

示例

示例 1：

输入：$\text{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}$
输出：$\text{[3,1,5]}$
解释：
这些餐馆为：
餐馆 $\text{1}$ $\text{[id=1, rating=4, veganFriendly=1, price=40, distance=10]}$
餐馆 $\text{2}$ $\text{[id=2, rating=8, veganFriendly=0, price=50, distance=5]}$
餐馆 $\text{3}$ $\text{[id=3, rating=8, veganFriendly=1, price=30, distance=4]}$
餐馆 $\text{4}$ $\text{[id=4, rating=10, veganFriendly=0, price=10, distance=3]}$
餐馆 $\text{5}$ $\text{[id=5, rating=1, veganFriendly=1, price=15, distance=1]}$
在按照 $\text{veganFriendly = 1, maxPrice = 50 和 maxDistance = 10}$ 进行过滤后，我们得到了餐馆 $\text{3}$、餐馆 $\text{1}$ 和餐馆 $\text{5}$（按评分从高到低排序）。

示例 2：

输入：$\text{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}$
输出：$\text{[4,3,2,1,5]}$
解释：餐馆与示例 1 相同，但在 $\text{veganFriendly = 0}$ 的过滤条件下，应该考虑所有餐馆。

示例 3：

输入：$\text{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}$
输出：$\text{[4,5]}$

数据范围

• $\text{1}\le \text{restaurants.length}\le {\text{10}}^{\text{4}}$
• $\text{restaurants[i].length}=\text{5}$
• $\text{1}\le {\text{id}}_{\text{i}}{\text{, rating}}_{\text{i}}{\text{, price}}_{\text{i}}{\text{, distance}}_{\text{i}}\le {\text{10}}^{\text{5}}$
• $\text{1}\le \text{maxPrice, maxDistance}\le {\text{10}}^{\text{5}}$
• ${\text{veganFriendly}}_{\text{i}}$ 和 $\text{veganFriendly}$ 的值为 $\text{0}$ 或 $\text{1}$
• 所有 ${\text{id}}_{\text{i}}$ 各不相同

解法

思路和算法

每个餐馆有五项信息：编号、评分、素食者友好、价格和距离，其中编号和评分用于排序，素食者友好、价格和距离用于过滤。由于只有符合三个过滤器条件的餐馆会包含在结果中，因此首先使用三个过滤器对餐馆过滤，排除不符合三个过滤器条件的餐馆，然后对符合三个过滤器条件的餐馆基于评分和编号降序排序。

对每个餐馆使用三个过滤器过滤的规则如下。

• 素食者友好过滤器：如果餐馆是素食者友好的，则无论给定的素食者友好条件 $\text{veganFriendly}$ 是 $1$ 还是 $0$，餐馆都符合条件；如果餐馆不是素食者友好的，则只有当给定的素食者友好条件 $\text{veganFriendly}$ 是 $0$ 的情况下，餐馆才符合条件。因此餐馆符合素食者友好过滤器条件等价于餐馆是素食者友好的或者给定的素食者友好条件 $\text{veganFriendly}$ 是 $0$。

• 最大价格过滤器：如果餐馆价格不超过给定的最大价格 $\text{maxPrice}$，则符合条件。

• 最大距离过滤器：如果餐馆距离不超过给定的最大距离 $\text{maxDistance}$，则符合条件。

创建排序列表存储符合三个过滤器条件的餐馆，排序列表用于对餐馆排序。由于排序的依据只有评分和编号，因此排序列表中只需要存储餐馆的编号和评分。将所有符合三个过滤器条件的餐馆加入排序列表之后，对列表排序，需要自定义如下排序规则。

• 如果两个餐馆的评分不同，则根据评分降序排序。

• 如果两个餐馆的评分相同，则根据编号降序排序。

排序之后，创建结果列表，依次将排序列表中的每个餐馆的编号添加到结果列表中，最后返回结果列表。

代码

class Solution {
    public List<Integer> filterRestaurants(int[][] restaurants, int veganFriendly, int maxPrice, int maxDistance) {
        List<int[]> restaurantList = new ArrayList<int[]>();
        for (int[] restaurant : restaurants) {
            if ((restaurant[2] == 1 || veganFriendly == 0) && restaurant[3] <= maxPrice && restaurant[4] <= maxDistance) {
                restaurantList.add(new int[]{restaurant[0], restaurant[1]});
            }
        }
        Collections.sort(restaurantList, (a, b) -> {
            if (a[1] != b[1]) {
                return b[1] - a[1];
            } else {
                return b[0] - a[0];
            }
        });
        List<Integer> filter = new ArrayList<Integer>();
        int size = restaurantList.size();
        for (int i = 0; i < size; i++) {
            int[] restaurant = restaurantList.get(i);
            filter.add(restaurant[0]);
        }
        return filter;
    }
}

复杂度分析

• 时间复杂度：$O(n\log n)$，其中 $n$ 是数组 $\text{restaurants}$ 的长度。过滤餐馆并将符合三个过滤器条件的餐馆添加到排序列表需要 $O(n)$ 的时间，对列表排序需要 $O(n\log n)$ 的时间，排序后将餐馆编号填入结果列表需要 $O(n)$ 的时间，时间复杂度是 $O(n\log n)$。

• 空间复杂度：$O(n)$，其中 $n$ 是数组 $\text{restaurants}$ 的长度。排序列表需要 $O(n)$ 的空间。