目录
题目:
示例:
分析:
代码:
题目:
示例:
分析:
题目挺长,估计中等难度是给在了阅读理解上。
简单来说就是给我们一堆餐厅的信息,每个餐厅拥有五个属性,分别是id,评分,是否是素食餐厅,价格和距离。
另外给我们三个过滤条件,分别是是否素食主义,最大价格和最远距离。
如果是素食主义那么就只能进素食餐厅,反之可以进所有餐厅。
另外只要价格和距离都没有超过最大的限制,那么也是可以的。
因此前面要做的只是按照题目的要求把符合条件的餐厅过滤出来。
最后要我们返回符合条件的餐厅id,按照评分的降序排序,评分一致时按照id的降序排序。
所以我们最后对过滤后的餐厅做个排序之后再把id提取出来就可以了。
代码:
class Solution {
public:
vector<int> filterRestaurants(vector<vector<int>>& restaurants, int veganFriendly, int maxPrice, int maxDistance) {
vector<vector<int>>cache;
for(const auto&r:restaurants){
//筛选出合格的餐厅
if((veganFriendly==0||r[2]==1) && r[4]<=maxDistance && r[3]<=maxPrice)
cache.push_back(r);
}
//按照要求排序
sort(cache.begin(),cache.end(),[](auto &a,auto &b){
if(a[1]==b[1]) return a[0]>b[0];
return a[1]>b[1];
});
vector<int>res(cache.size());
//按顺序取出餐厅id
for(int i=0;i<cache.size();++i) res[i]=cache[i][0];
return res;
}
};