0数据结构设计/哈希 LeetCode2353. 设计食物评分系统

2353. 设计食物评分系统

描述

设计一个支持下述操作的食物评分系统：

修改 系统中列出的某种食物的评分。
返回系统中某一类烹饪方式下评分最高的食物。
实现 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] 之前。

来源：力扣（LeetCode）
链接：https://leetcode.cn/problems/design-a-food-rating-system
著作权归领扣网络所有。商业转载请联系官方授权，非商业转载请注明出处。

分析

使用两个map实现功能，第一个map映射food与rating,cuisine；第二个map映射cuisine与food,rating的有序set集合，方便快速找到rating最大的food。

class FoodRatings {
Map<String, Pair<String,Integer>> mapf = new HashMap<>();Map<String, TreeSet<Pair<String,Integer>>> mapc = new HashMap<>();
    public FoodRatings(String[] foods, String[] cuisines, int[] ratings) {
for (int i = 0; i < foods.length; i++) {
            Pair<String,Integer> mapfTmp = new Pair<>(cuisines[i],ratings[i]);
            mapf.put(foods[i],mapfTmp);

            if (mapc.get( cuisines[i]) == null) {
                TreeSet<Pair<String,Integer>> treeSet = new TreeSet<>((a,b)->{
                    if (!Objects.equals(a.getValue(),b.getValue())) {
                        return b.getValue() - a.getValue();
                    } else {
                        return a.getKey().compareTo(b.getKey());
                    }
                });
                mapc.put(cuisines[i],treeSet);
            }
            TreeSet<Pair<String,Integer>> set = mapc.get(cuisines[i]);
            Pair<String,Integer> mapcTmp = new Pair<>(foods[i],ratings[i]);
            set.add(mapcTmp);
            mapc.put(cuisines[i],set);
        }
    }
    
    public void changeRating(String food, int newRating) {
Pair<String,Integer> mapftmp = mapf.get(food);
        String cus = mapftmp.getKey();
        mapf.put(food,new Pair<>(cus,newRating));
        TreeSet<Pair<String,Integer>> treeSet = mapc.get(cus);
        treeSet.remove(new Pair(food,mapftmp.getValue()));
        treeSet.add(new Pair(food,newRating));
    }
    
    public String highestRated(String cuisine) {
return mapc.get(cuisine).first().getKey();
    }
}

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