- [1964 Find the Longest Valid Obstacle Course at Each Position](https://leetcode.com/problems/find-the-longest-valid-obstacle-course-at-each-position/description/)

- [2140 Solving Questions With Brainpower](https://leetcode.com/problems/solving-questions-with-brainpower/description/)

- [2215 Find the Difference of Two Arrays](https://leetcode.com/problems/find-the-difference-of-two-arrays/description/)

- [2353 Design a Food Rating System](https://leetcode.com/problems/design-a-food-rating-system/description/)

- [2390 Removing Stars From a String](https://leetcode.com/problems/removing-stars-from-a-string/description/)

- [2466 Count Ways To Build Good Strings](https://leetcode.com/problems/count-ways-to-build-good-strings/description/)

- [2523 Closest Prime Numbers in Range](https://leetcode.com/problems/closest-prime-numbers-in-range/description/)

import collections

import heapq

from dataclasses import dataclass, field

from typing import List

class Food:

"""Item stored in the priority queue."""

rating: int = field(compare=True)

food: str = field(compare=True)

cuisine: str = field(compare=False)

class FoodRatings:

"""

def __init__(self, foods: List[str], cuisines: List[str], ratings: List[int]):

self.food_to_cuisine = {}

self.food_to_rating = {}

self.cuisine_to_max_food = collections.defaultdict(list)

for f, c, r in zip(foods, cuisines, ratings, strict=False):

self.food_to_cuisine[f] = c

self.food_to_rating[f] = -r

heapq.heappush(self.cuisine_to_max_food[c], Food(-r, f, c))

def changeRating(self, food: str, newRating: int) -> None:

# Update food_to_rating

self.food_to_rating[food] = -newRating

# Insert the new food element in the priority queue

cuisine = self.food_to_cuisine[food]

heapq.heappush(

self.cuisine_to_max_food[cuisine], Food(-newRating, food, cuisine)

)

def highestRated(self, cuisine: str) -> str:

# Get the highest rated food of

max_food = self.cuisine_to_max_food[cuisine][0]

while self.food_to_rating[max_food.food] != max_food.rating:

heapq.heappop(self.cuisine_to_max_food[cuisine])

max_food = self.cuisine_to_max_food[cuisine][0]

return max_food.food

from awesome_python_leetcode._2353_design_a_food_rating_system import FoodRatings

def test_func():

"""Tests the solution of a LeetCode problem."""

food_ratings = FoodRatings(

foods=["kimchi", "miso", "sushi", "moussaka", "ramen", "bulgogi"],

cuisines=["korean", "japanese", "japanese", "greek", "japanese", "korean"],

ratings=[9, 12, 8, 15, 14, 7],

)

assert food_ratings.highestRated("korean") == "kimchi"

assert food_ratings.highestRated("japanese") == "ramen"

food_ratings.changeRating("sushi", 16)

assert food_ratings.highestRated("japanese") == "sushi"

food_ratings.changeRating("ramen", 16)

assert food_ratings.highestRated("japanese") == "ramen"