Python SortedSet 实践：如何安全地更新排序元素的键值

from sortedcontainers import SortedSet
from typing import List

class FoodRatings:
    def __init__(self, foods: List[str], cuisines: List[str], ratings: List[int]):
        self.food_map = {}  # 存储食物详情: Food: [cuisine, rating, food_name]
        self.cuisines_map = {}  # 存储菜系对应的SortedSet: Cuisine: SortedSet(Food)

        for index in range(len(foods)):
            food_name = foods[index]
            cuisine = cuisines[index]
            rating = ratings[index]

            self.food_map[food_name] = [cuisine, rating, food_name]

            if cuisine not in self.cuisines_map:
                # 初始化SortedSet，指定排序键
                # 评分降序，食物名称字典序升序
                self.cuisines_map[cuisine] = SortedSet(key=lambda x: (-self.food_map[x][1], self.food_map[x][2]))

            self.cuisines_map[cuisine].add(food_name)

    def highestRated(self, cuisine: str) -> str:
        """返回指定菜系中评分最高的食物（或字典序最小的）"""
        if cuisine in self.cuisines_map and self.cuisines_map[cuisine]:
            return self.cuisines_map[cuisine][0]
        return None # 或抛出异常，视具体业务逻辑而定

    def changeRating_problematic(self, food: str, newRating: int) -> None:
        cuisine = self.food_map[food][0]

        # 1. 直接修改评分，此时食物仍在SortedSet中
        self.food_map[food][1] = newRating 

        # 2. 尝试从SortedSet中移除食物
        # 此时，'food'在SortedSet中的“旧键”对应的位置已不再是其“新键”对应的位置
        # 导致SortedSet无法找到并移除该元素
        self.cuisines_map[cuisine].discard(food) # 可能报错或行为异常

        # 3. 重新添加食物
        self.cuisines_map[cuisine].add(food)

    def changeRating(self, food: str, newRating: int) -> None:
        cuisine = self.food_map[food][0]

        # 1. 首先从SortedSet中移除食物。
        # 此时'food'的评分尚未修改，SortedSet可以根据其当前的（旧的）键值正确找到并移除。
        self.cuisines_map[cuisine].discard(food)

        # 2. 修改食物的评分。
        # 此时'food'已不在SortedSet中，修改其属性是安全的。
        self.food_map[food][1] = newRating

        # 3. 重新将食物添加到SortedSet中。
        # SortedSet会根据'food'当前（新的）评分重新计算键值，并将其插入到正确的位置。
        self.cuisines_map[cuisine].add(food)

# 示例代码重现问题与解决方案
obj = FoodRatings(["kimchi","miso","sushi","moussaka","ramen","bulgogi"],
                  ["korean","japanese","japanese","greek","japanese","korean"],
                  [9,12,8,15,14,7])

# 使用正确的changeRating方法
obj.changeRating("sushi", 16) 
# 此时，"sushi"的评分已更新为16，并且在japanese菜系的SortedSet中正确排序。
print(obj.highestRated("japanese")) # 应该输出 "sushi" (16分最高)