算法|每日一题|餐厅过滤器

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 。

如果题目不是很清晰，可以到力扣去看示例

class Solution {
    public List<Integer> filterRestaurants(int[][] restaurants, int veganFriendly, int maxPrice, int maxDistance) {
        //判断veganFriendly是否为true
        //length记录过滤后剩下的店铺数量
        int length1 = 0;
        for(int i = 0; i < restaurants.length; i++){
            if(restaurants[i][3] <= maxPrice && restaurants[i][4] <= maxDistance){
                restaurants[length1++] = restaurants[i];
            }
        }
        int length2 = 0;
        if(veganFriendly == 1){
            for(int  j = 0; j < length1; j++){
                if(restaurants[j][2] == 1){
                    restaurants[length2++] = restaurants[j];
                }
            }
        }else{
            length2 = length1;
        }

        List<Integer> list = new ArrayList<Integer>();

        //选择排序将rating最大的放到左边
        //rating相同时比较id，这道题要求id数值大的在前边
        for(int m = 0; m < length2 - 1; m++){
            int max = m;
            for(int n = m + 1; n < length2; n++){
                if(restaurants[max][1] < restaurants[n][1]){
                    max = n;
                }else if(restaurants[max][1] == restaurants[n][1]){
                    if(restaurants[max][0] < restaurants[n][0]){
                        max = n;
                    }
                }
            }
            int[] temp = restaurants[m];
            restaurants[m] = restaurants[max];
            restaurants[max] = temp;
            list.add(restaurants[m][0]);
        }
        //添加最后一个餐馆
        if(length2 != 0){
            list.add(restaurants[length2 - 1][0]);
        }
        return list;
    }
}

如果对您有帮助，请点赞关注支持我，谢谢！
如有错误或者不足之处，希望您能够加以指正！