题目链接
Leetcode.2353 设计食物评分系统 Rating : 1782
题目描述
设计一个支持下述操作的食物评分系统:
-
修改 系统中列出的某种食物的评分。
-
返回系统中某一类烹饪方式下评分最高的食物。
实现FoodRatings
类: -
FoodRatings(String[] foods, String[] cuisines, int[] ratings)
初始化系统。食物由foods、cuisines
和ratings
描述,长度均为n
。 -
foods[i]
是第i
种食物的名字。
-
cuisines[i]
是第i
种食物的烹饪方式。
-
ratings[i]
是第i
种食物的最初评分。
-
void changeRating(String food, int newRating)
修改名字为food
的食物的评分。 -
String highestRated(String cuisine)
返回指定烹饪方式cuisine
下评分最高的食物的名字。如果存在并列,返回 字典序较小 的名字。
注意,字符串x
的字典序比字符串y
更小的前提是:x
在字典中出现的位置在y
之前,也就是说,要么x
是y
的前缀,或者在满足x[i] != y[i]
的第一个位置i
处,x[i]
在字母表中出现的位置在y[i]
之前。
示例:
输入
[“FoodRatings”, “highestRated”, “highestRated”, “changeRating”, “highestRated”, “changeRating”, “highestRated”]
[[[“kimchi”, “miso”, “sushi”, “moussaka”, “ramen”, “bulgogi”], [“korean”, “japanese”, “japanese”, “greek”, “japanese”, “korean”], [9, 12, 8, 15, 14, 7]], [“korean”], [“japanese”], [“sushi”, 16], [“japanese”], [“ramen”, 16], [“japanese”]]
输出
[null, “kimchi”, “ramen”, null, “sushi”, null, “ramen”]解释 FoodRatings foodRatings = new FoodRatings([“kimchi”, “miso”,
“sushi”, “moussaka”, “ramen”, “bulgogi”], [“korean”, “japanese”,
“japanese”, “greek”, “japanese”, “korean”], [9, 12, 8, 15, 14, 7]);
foodRatings.highestRated(“korean”); // 返回 “kimchi”
// “kimchi” 是分数最高的韩式料理,评分为 9 。 foodRatings.highestRated(“japanese”); // 返回 “ramen”
// “ramen” 是分数最高的日式料理,评分为 14 。 foodRatings.changeRating(“sushi”, 16); // “sushi” 现在评分变更为 16 。
foodRatings.highestRated(“japanese”); // 返回 “sushi”
// “sushi” 是分数最高的日式料理,评分为 16 。 foodRatings.changeRating(“ramen”, 16); // “ramen” 现在评分变更为 16 。
foodRatings.highestRated(“japanese”); // 返回 “ramen”
// “sushi” 和 “ramen” 的评分都是 16 。
// 但是,“ramen” 的字典序比 “sushi” 更小。
提示:
- 1 < = n < = 2 ∗ 1 0 4 1 <= n <= 2 * 10^4 1<=n<=2∗104
- n = = f o o d s . l e n g t h = = c u i s i n e s . l e n g t h = = r a t i n g s . l e n g t h n == foods.length == cuisines.length == ratings.length n==foods.length==cuisines.length==ratings.length
- 1 < = f o o d s [ i ] . l e n g t h , c u i s i n e s [ i ] . l e n g t h < = 10 1 <= foods[i].length, cuisines[i].length <= 10 1<=foods[i].length,cuisines[i].length<=10
foods[i]、cuisines[i]
由小写英文字母组成- 1 < = r a t i n g s [ i ] < = 1 0 8 1 <= ratings[i] <= 10^8 1<=ratings[i]<=108
foods
中的所有字符串 互不相同- 在对
changeRating
的所有调用中,food
是系统中食物的名字。 - 在对
highestRated
的所有调用中,cuisine
是系统中 至少一种 食物的烹饪方式。 - 最多调用
changeRating
和highestRated
总计 2 ∗ 1 0 4 2 * 10^4 2∗104 次
分析:
用两个 哈希表 分别记录:
unordered_map<string,pair<string,int>>
建立food -> (cuisine,rating)
的映射关系unordered_map<string,set<pair<int,string>>>
建立cuisine -> rating最高的,food字典序最小的映射关系 (-rating,food)
因为 set
是有序列表,类似于Java
的TreeSet
,它会自动把里面的元素按 从小到大
的顺序排序。如果里面的元素是pair
,那么他会先按 pair的第一个元素
排序,接着再按 pair的第二个元素
排序。
所以我们只需要 按(-rating,food)
插入元素,最后set
的第一个元素就是 Rating最高的,food字典序最小的 元素。
代码:
class FoodRatings {
public:
int n;
// food ->(cuisine,rating)
unordered_map<string,pair<string,int>> cf;
//cuisine -> set(-rating,food)
unordered_map<string,set<pair<int,string>>> fs;
FoodRatings(vector<string>& foods, vector<string>& cuisines, vector<int>& ratings) {
this->n = foods.size();
//建立各自的映射关系
for(int i = 0;i < n;i++){
auto food = foods[i];
auto c = cuisines[i];
auto r = ratings[i];
cf[food] = {c,r};
fs[c].emplace(-r,food);
}
}
void changeRating(string food, int newRating) {
auto &[c,r] = cf[food];
//删除旧的 -rating 记录
fs[c].erase({-r,food});
//插入新的 -newRating 记录
fs[c].emplace(-newRating,food);
//将 旧的r 改为 新的newRating
r = newRating;
}
string highestRated(string cuisine) {
//set中第一个元素 的 second 就是rating最高的,food字典序最小的
return fs[cuisine].begin()->second;
}
};
/**
* Your FoodRatings object will be instantiated and called as such:
* FoodRatings* obj = new FoodRatings(foods, cuisines, ratings);
* obj->changeRating(food,newRating);
* string param_2 = obj->highestRated(cuisine);
*/