【leetcode 每日打卡】2034. 股票价格波动

在这里插入图片描述

from sortedcontainers import SortedList

class StockPrice:
    def __init__(self):
        self.price = SortedList()
        self.timePriceMap = {}
        self.maxTimestamp = 0

    def update(self, timestamp: int, price: int) -> None:
        if timestamp in self.timePriceMap:
            self.price.discard(self.timePriceMap[timestamp])
        self.price.add(price)
        self.timePriceMap[timestamp] = price
        self.maxTimestamp = max(self.maxTimestamp, timestamp)

    def current(self) -> int:
        return self.timePriceMap[self.maxTimestamp]

    def maximum(self) -> int:
        return self.price[-1]

    def minimum(self) -> int:
        return self.price[0]

1.SortedList()是python中一个专门提供排序的库,通过add()增加,discard()删除(即使不存在数据也不会报错),增加到数组的数字会从小到大自动排序,这样就可以通过[0]来访问数组最小值,[-1]来访问数组最大值。

2.首先初始化类,price代表一个从小到大自动排序的价格表,timePriceMap存放的是时间点所对应的价格,maxTimestamp代表是当前的时间,也就是最大的时间戳。

3.先判断当前时间戳是否已经存在,如果存在就删除内容,然后就将当前时间戳所对应的价格存入timePriceMap的字典中,并且将价格放入SortedList自动排序,通过max()和当前时间戳的大小进行比对从而进行更新最新时间。

评论 1
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值