题目链接
Leetcode.2034 股票价格波动
rating : 1832
题目描述
给你一支股票价格的数据流。数据流中每一条记录包含一个 时间戳 和该时间点股票对应的 价格 。
不巧的是,由于股票市场内在的波动性,股票价格记录可能不是按时间顺序到来的。某些情况下,有的记录可能是错的。如果两个有相同时间戳的记录出现在数据流中,前一条记录视为错误记录,后出现的记录 更正 前一条错误的记录。
请你设计一个算法,实现:
- 更新 股票在某一时间戳的股票价格,如果有之前同一时间戳的价格,这一操作将 更正 之前的错误价格。
- 找到当前记录里 最新股票价格 。最新股票价格 定义为时间戳最晚的股票价格。
- 找到当前记录里股票的 最高价格 。
- 找到当前记录里股票的 最低价格 。
请你实现 S t o c k P r i c e StockPrice 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 ≤ t i m e s t a m p , p r i c e ≤ 1 0 9 1 \leq timestamp, price \leq 10^9 1≤timestamp,price≤109
update
,current
,maximum
和minimum
总 调用次数不超过 1 0 5 10^5 105 。current
,maximum
和minimum
被调用时,update
操作 至少 已经被调用过 一次 。
解法:哈希表 + 模拟
用一个哈希表 m m m 来记录,股票时间戳 和 价格的关系,即 [ 时间戳 , 价格 ]
。
用另外一个哈希表 p p p 来记录,同一个价格下的不同时间戳,即 [ 价格1 : [时间戳1 , 时间戳2 , 时间戳3] ]
。
m m m 和 p p p 都是 自动按照键值从小到大排序的。
这样我们就能快速获取 当前股票的价格,最高价格,最低价格。
我们只需要维护这两个哈希表即可。
时间复杂度:current
,maximum
, minimum
和 update
操作都是 O ( 1 ) O(1) O(1)
C++代码:
class StockPrice {
public:map<int,int> m;map<int,unordered_set<int>> p;StockPrice() {}void update(int timestamp, int price) {if(!m.count(timestamp)){m[timestamp] = price;p[price].insert(timestamp);}else{int old_price = m[timestamp];p[old_price].erase(timestamp);if(p[old_price].size() == 0) p.erase(old_price);m[timestamp] = price;p[price].insert(timestamp);}}int current() {auto it = m.end();--it;return it->second;}int maximum() {auto it = p.end();--it;return it->first;}int minimum() {auto it = p.begin();return it->first;}
};/*** Your StockPrice object will be instantiated and called as such:* StockPrice* obj = new StockPrice();* obj->update(timestamp,price);* int param_2 = obj->current();* int param_3 = obj->maximum();* int param_4 = obj->minimum();*/