[Leetcode]1333.餐厅过滤器

题目：给你一个餐馆信息数组 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 。

示例：

输入：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（按评分从高到低排序）。

---

思路：

1. 先根据过滤器，排除不符合过滤器条件的餐馆；

   1. 注意veganFriendly是0的情况下是所有餐馆都符合条件。

2. 对符合过滤器条件餐馆基于评分rating降序排序；

3. 如果rating相同，再基于其编号id降序排序。

//题解
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;
    }
}

//评论与题解同样思路不同写法
class Solution {
    public List<Integer> filterRestaurants(int[][] restaurants, int veganFriendly, int maxPrice, int maxDistance) {
        // 过滤收集
        List<int[]> collector = new ArrayList<>();
        for (int[] restaurant : restaurants) {
            if (veganFriendly == 1 && restaurant[2] == 0 || restaurant[3] > maxPrice || restaurant[4] > maxDistance) {
                continue;
            }
            collector.add(restaurant);
        }
        // 排序
        collector.sort((o1, o2) -> o1[1] == o2[1] ? o2[0] - o1[0] : o2[1] - o1[1]);
        List<Integer> res = new ArrayList<>();
        for (int[] rest : collector) {
            res.add(rest[0]);
        }
        return res;
    }
}

---

• List<int[]> collector = new ArrayList<>();

  • List<int[]>可以转二维数组。

  • List是一个接口，<>表示List里面放的对象是什么类型的，

  • ArrayList类是继承AbstractList抽象类和实现List接口的一个实现类。

    因此，List接口不能被构造，也就是我们说的不能创建实例对象，但是我们可以像下面那样为List接口创建一个指向自己的对象引用，而ArrayList实现类的实例对象就在这充当了这个指向List接口的对象引用。