菜鸟的每日力扣系列——2034. 股票价格波动

from sortedcontainers import SortedList


class StockPrice:
    def __init__(self):
        self.max_time = 0
        self.price = SortedList()  # 使用有序列表存放价格
        self.res = {}

    def update(self, timestamp: int, price: int) -> None:
        if timestamp in self.res:  # 如果结果集中存放了时间戳
            self.price.remove(self.res[timestamp])  # 结果集删除掉哈希表对应的结果
        self.price.add(price)  # 存入新的价格
        self.res[timestamp] = price  # 将新价格及时间戳放入哈希表
        self.max_time = max(self.max_time, timestamp)

    def current(self) -> int:
        return self.res[self.max_time]

    def maximum(self) -> int:
        return self.price[-1]  # 有序列表中最后一位即为最大值

    def minimum(self) -> int:
        return self.price[0]  # 有序列表中第一位即为最大值


stockPrice = StockPrice()
stockPrice.update(1, 10)
stockPrice.update(2, 5)
print(stockPrice.current())  # 5
print(stockPrice.maximum())  # 10
stockPrice.update(1, 3)
print(stockPrice.maximum())  # 5
stockPrice.update(4, 2)
print(stockPrice.minimum())  # 2