LeetCode刷题日记2022-1-23/2034. 股票价格波动-哈希表&有序集合
2034. 股票价格波动
- 题目描述
- 题解思路
-
- SortedList
- 题解代码
题目描述
给你一支股票价格的数据流。数据流中每一条记录包含一个 时间戳 和该时间点股票对应的 价格 。
不巧的是，由于股票市场内在的波动性，股票价格记录可能不是按时间顺序到来的。某些情况下，有的记录可能是错的。如果两个有相同时间戳的记录出现在数据流中，前一条记录视为错误记录，后出现的记录 更正 前一条错误的记录。
请你设计一个算法，实现：
- 更新 股票在某一时间戳的股票价格，如果有之前同一时间戳的价格，这一操作将 更正 之前的错误价格。
- 找到当前记录里 最新股票价格 。最新股票价格 定义为时间戳最晚的股票价格。
- 找到当前记录里股票的 最高价格 。
- 找到当前记录里股票的 最低价格 。
请你实现 StockPrice 类：
- StockPrice() 初始化对象，当前无股票价格记录。
- void update(int timestamp, int price) 在时间点 timestamp 更新股票价格为 price 。
- int current() 返回股票 最新价格 。
- int maximum() 返回股票 最高价格 。
- int minimum() 返回股票 最低价格 。
示例 1：
输入：
["StockPrice", "update", "update", "current", "maximum", "update", "maximum", "update", "minimum"]
[[], [1, 10], [2, 5], [], [], [1, 3], [], [4, 2], []]
输出：
[null, null, null, 5, 10, null, 5, null, 2]
解释：
StockPrice stockPrice = new StockPrice();
stockPrice.update(1, 10); // 时间戳为 [1] ，对应的股票价格为 [10] 。
stockPrice.update(2, 5); // 时间戳为 [1,2] ，对应的股票价格为 [10,5] 。
stockPrice.current(); // 返回 5 ，最新时间戳为 2 ，对应价格为 5 。
stockPrice.maximum(); // 返回 10 ，最高价格的时间戳为 1 ，价格为 10 。
stockPrice.update(1, 3); // 之前时间戳为 1 的价格错误，价格更新为 3 。
// 时间戳为 [1,2] ，对应股票价格为 [3,5] 。
stockPrice.maximum(); // 返回 5 ，更正后最高价格为 5 。
stockPrice.update(4, 2); // 时间戳为 [1,2,4] ，对应价格为 [3,5,2] 。
stockPrice.minimum(); // 返回 2 ，最低价格时间戳为 4 ，价格为 2 。
提示：
- 1 <= timestamp, price <= 109
- update，current，maximum 和 minimum 总 调用次数不超过 105 。
- current，maximum 和 minimum 被调用时，update 操作 至少 已经被调用过 一次 。
题解思路
因为题目要求我们要更新某一时间戳得价格 如果存在的话则修正 如果不存在则添加 因此对于这一部分的更新我们采用
哈希表
其次 题目还要求我们返回当前记录的 股票价格最高和最低价格 因此我们采用有序集合（SortedList） 来记录股票的价
格
SortedList
使用参考：https://www.jb51.net/article/221847.htm
题解代码
from sortedcontainers import SortedList
class StockPrice:
def __init__(self):
self.price=SotredList()
self.timePriceMap={}
self.maxTimetamp=0
def update(self, timestamp: int, price: int) -> None:
if timestamp in self.timePriceMap:
self.price.discard(self.timePriceMap[timestamp])
self.timePriceMap[timestamp]=price
self.price.add(price)
self.maxTimetamp=max(self.maxTimetamp,timestamp)
def current(self) -> int:
return self.timePriceMap[self.maxTimetamp]
def maximum(self) -> int:
return self.price[-1]
def minimum(self) -> int:
return self.price[0]