1. 题目描述
设计一个支持下述操作的食物评分系统:
修改 系统中列出的某种食物的评分。
返回系统中某一类烹饪方式下评分最高的食物。
实现 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 * 104
n == foods.length = cuisines.length == ratings.length
1 <= foods[i].length, cuisines[i].length <= 10
foods[i]、cuisines[i] 由小写英文字母组成
1 <= ratings[i] <= 108
foods 中的所有字符串 互不相同
在对 changeRating 的所有调用中,food 是系统中食物的名字。
在对 highestRated 的所有调用中,cuisine 是系统中 至少一种 食物的烹饪方式。
最多调用 changeRating 和 highestRated 总计 2 * 104 次
2. 分析
核心思路:哈希表+大根堆。
同时维护大根堆内元素所处索引的记忆化
3. 代码
type Pair struct {
// 映射表中的索引
idx int
food string
cuisine string
rating int
}
type hp []*Pair
func (hp hp) Len() int { return len(hp) }
func (hp hp) Swap(i, j int) {
// 同步更新索引位置
hp[i].idx, hp[j].idx = j, i
hp[i], hp[j] = hp[j], hp[i]
}
func (hp hp) Less(i, j int) bool {
if hp[i].rating == hp[j].rating {
return hp[i].food < hp[j].food
}
return hp[i].rating > hp[j].rating
}
// 没有pop操作,可以做空实现
func (hp *hp) Pop() any { return nil }
func (hp *hp) Push(v any) {
p := v.(*Pair)
p.idx = len(*hp)
*hp = append(*hp, p)
}
type FoodRatings struct {
// food -> {food, rating}
foods map[string]*Pair
// cuisine -> 最大堆
cuisines map[string]*hp
}
func Constructor(foods []string, cuisines []string, ratings []int) FoodRatings {
fds := map[string]*Pair{}
css := map[string]*hp{}
for i := range foods {
food := foods[i]
cuisine := cuisines[i]
rating := ratings[i]
p := &Pair{food: food, cuisine: cuisine, rating: rating}
if css[cuisine] == nil {
css[cuisine] = &hp{}
}
fds[food] = p
heap.Push(css[cuisine], p)
}
return FoodRatings{
foods: fds,
cuisines: css,
}
}
func (this *FoodRatings) ChangeRating(food string, newRating int) {
p := this.foods[food]
p.rating = newRating
heap.Fix(this.cuisines[p.cuisine], p.idx)
}
func (this *FoodRatings) HighestRated(cuisine string) string {
hp := this.cuisines[cuisine]
return (*hp)[0].food
}
评论