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\ast 10^4$
• $n==foods.length==cuisines.length==ratings.length$
• $1<=foods[i].length,cuisines[i].length<=10$
• foods[i]、cuisines[i]由小写英文字母组成
• $1<=ratings[i]<=10^8$
• foods中的所有字符串 互不相同
• 在对 changeRating的所有调用中，food是系统中食物的名字。
• 在对 highestRated的所有调用中，cuisine是系统中 至少一种 食物的烹饪方式。
• 最多调用 changeRating和 highestRated总计 $2\ast 10^4$ 次

分析：

用两个 哈希表 分别记录：

• 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);
 */