leetcode地址
给你一个餐馆信息数组 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 的过滤条件下,应该考虑所有餐馆。
//存储满足条件的餐厅
List 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() {
@Override
public int compare(Integer[] o1, Integer[] o2) {
int ans = 0;
if (o1[1]>o2[1]){//根据rate排序
ans=-1;
}else if (o1[1]o2[0]){ //rate相同根据id排序
ans=-1;
}else if (o1[0]
List ans = new ArrayList<>();
//3. 取出满足条件的餐厅id
for (int i = 0; i < list.size(); i++) {
ans.add(list.get(i)[0]);
}
class Solution {
public List filterRestaurants(int[][] restaurants,
int veganFriendly,
int maxPrice,
int maxDistance) {
//存储满足条件的餐厅
List 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() {
@Override
public int compare(Integer[] o1, Integer[] o2) {
int ans = 0;
if (o1[1]>o2[1]){
ans=-1;
}else if (o1[1]o2[0]){
ans=-1;
}else if (o1[0] ans = new ArrayList<>();
//3. 取出满足条件的餐厅id
for (int i = 0; i < list.size(); i++) {
ans.add(list.get(i)[0]);
}
return ans;
}
}