【每日一题Day339】LC1333餐厅过滤器 | 排序

餐厅过滤器【LC1333】

给你一个餐馆信息数组 restaurants,其中 restaurants[i] = [idi, ratingi, veganFriendlyi, pricei, distancei]。你必须使用以下三个过滤器来过滤这些餐馆信息。

其中素食者友好过滤器 veganFriendly 的值可以为 true 或者 false,如果为 true 就意味着你应该只包括 veganFriendlyi 为 true 的餐馆,为 false 则意味着可以包括任何餐馆。此外,我们还有最大价格 maxPrice 和最大距离 maxDistance 两个过滤器,它们分别考虑餐厅的价格因素和距离因素的最大值。

过滤后返回餐馆的 *id*,按照 rating 从高到低排序。如果 rating 相同,那么按 id 从高到低排序。简单起见, veganFriendlyiveganFriendlytrue 时取值为 1,为 false 时,取值为 0 。

  • 思路

    根据条件筛选餐厅,然后将餐厅按照规则排序,最后将id按照顺序放入结果集中

  • 实现

    class Solution {
        public List<Integer> filterRestaurants(int[][] restaurants, int veganFriendly, int maxPrice, int maxDistance) {
            List<int[]> list = new ArrayList<>();
            for (int[] r : restaurants){
                if (r[2] >= veganFriendly && r[3] <= maxPrice && r[4] <= maxDistance){
                    list.add(new int[]{r[0], r[1]});
                }
            }
            list.sort((o1, o2) -> o1[1] == o2[1] ? o2[0] - o1[0] : o2[1] - o1[1]);
            List<Integer> res = new ArrayList<>();
            for (int[] data : list){
                res.add(data[0]);
            }
            return res;
        }
    }
    
    • 复杂度
      • 时间复杂度: O ( n ∗ log ⁡ n ) \mathcal{O}(n*\log n) O(nlogn)
      • 空间复杂度: O ( log ⁡ n ) \mathcal{O}(\log n) O(logn)

你可能感兴趣的:(每日一题,排序,leetcode)