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;
}
}
如果对您有帮助,请点赞关注支持我,谢谢!
如有错误或者不足之处,希望您能够加以指正!