Leetcode.2353 设计食物评分系统
题目链接
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∗1041 <= n <= 2 * 10^41<=n<=2∗104
- n==foods.length==cuisines.length==ratings.lengthn == foods.length == cuisines.length == ratings.lengthn==foods.length==cuisines.length==ratings.length
- 1<=foods[i].length,cuisines[i].length<=101 <= foods[i].length, cuisines[i].length <= 101<=foods[i].length,cuisines[i].length<=10
foods[i]、cuisines[i]
由小写英文字母组成- 1<=ratings[i]<=1081 <= ratings[i] <= 10^81<=ratings[i]<=108
foods
中的所有字符串 互不相同- 在对
changeRating
的所有调用中,food
是系统中食物的名字。 - 在对
highestRated
的所有调用中,cuisine
是系统中 至少一种 食物的烹饪方式。 - 最多调用
changeRating
和highestRated
总计 2∗1042 * 10^42∗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 改为 新的newRatingr = 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);*/