leetCode 1333. 餐厅过滤器

 leetcode地址

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 的过滤条件下，应该考虑所有餐馆。

2.解题方法 

1. 解题方法按素食、距离和价格筛选符合条件的餐厅。

//存储满足条件的餐厅
List<Integer[]> list = new ArrayList<>();
//1.遍历餐厅，筛选满足条件的餐厅
for (int i = 0; i < restaurants.length; i++) {
  if (maxPrice>=restaurants[i][3]&&maxDistance>=restaurants[i][4]){
       int length = restaurants[i].length;
      Integer[] arr = new Integer[length];
      for (int j = 0; j < length; j++) {
          arr[j] = restaurants[i][j];
      }

      if (veganFriendly==1&&restaurants[i][2]==1){//素食
          list.add(arr);
      }else if (veganFriendly==0){
          list.add(arr);
      }
  }
}

    2.对筛选出的餐厅进行排序。

 //2.自定义排序规则 按照 rating 从高到低排序。
//如果 rating 相同，那么按 id 从高到低排序
Collections.sort(list, new Comparator<Integer[]>() {
    @Override
    public int compare(Integer[] o1, Integer[] o2) {
    int ans = 0;
    if (o1[1]>o2[1]){//根据rate排序
         ans=-1;
    }else if (o1[1]<o2[1]){
         ans=1;
    }else {
     if (o1[0]>o2[0]){ //rate相同根据id排序
        ans=-1;
        }else if (o1[0]<o2[0]){
        ans=1;
        }
     }
    return ans;
    }
});

3. 取出排序和餐厅的id并返回。

List<Integer> ans = new ArrayList<>();
//3. 取出满足条件的餐厅id
for (int i = 0; i < list.size(); i++) {
  ans.add(list.get(i)[0]);
}

3.复杂度

• 空间复杂度: O(n)
• 时间复杂度: O(nlogn) 主要是排序耗时

4.Code

class Solution {
public  List<Integer> filterRestaurants(int[][] restaurants,
                                         int veganFriendly,
                                         int maxPrice,
                                         int maxDistance) {
      //存储满足条件的餐厅
      List<Integer[]> list = new ArrayList<>();
      //1.遍历餐厅，筛选满足条件的餐厅
      for (int i = 0; i < restaurants.length; i++) {
          if (maxPrice>=restaurants[i][3]&&maxDistance>=restaurants[i][4]){
              int length = restaurants[i].length;
              Integer[] arr = new Integer[length];
              for (int j = 0; j < length; j++) {
                  arr[j] = restaurants[i][j];
              }
              if (veganFriendly==1&&restaurants[i][2]==1){
                  list.add(arr);
              }else if (veganFriendly==0){
                  list.add(arr);
              }
          }
      }
      //2.自定义排序规则 按照 rating 从高到低排序。
      //如果 rating 相同，那么按 id 从高到低排序
      Collections.sort(list, new Comparator<Integer[]>() {
          @Override
          public int compare(Integer[] o1, Integer[] o2) {
              int ans = 0;
              if (o1[1]>o2[1]){
                  ans=-1;
              }else if (o1[1]<o2[1]){
                  ans=1;
              }else {
                  if (o1[0]>o2[0]){
                      ans=-1;
                  }else if (o1[0]<o2[0]){
                      ans=1;
                  }
              }
              return ans;
          }
      });
      
      List<Integer> ans = new ArrayList<>();
      //3. 取出满足条件的餐厅id
      for (int i = 0; i < list.size(); i++) {
          ans.add(list.get(i)[0]);
      }
      return ans;
  }
}

5.执行效果