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

餐厅过滤器【LC1333】

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

• 思路

  根据条件筛选餐厅，然后将餐厅按照规则排序，最后将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;
      }
  }
  

  • 复杂度
    • 时间复杂度：$\mathcal{O}(n\ast \log n)$
    • 空间复杂度：$\mathcal{O}(\log n)$