题目
给你一支股票价格的数据流。数据流中每一条记录包含一个 时间戳 和该时间点股票对应的 价格 。
不巧的是,由于股票市场内在的波动性,股票价格记录可能不是按时间顺序到来的。某些情况下,有的记录可能是错的。如果两个有相同时间戳的记录出现在数据流中,前一条记录视为错误记录,后出现的记录 更正 前一条错误的记录。
请你设计一个算法,实现:
- 更新 股票在某一时间戳的股票价格,如果有之前同一时间戳的价格,这一操作将 更正 之前的错误价格。
- 找到当前记录里 最新股票价格 。最新股票价格 定义为时间戳最晚的股票价格。
- 找到当前记录里股票的 最高价格 。
- 找到当前记录里股票的 最低价格 。
请你实现 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 。
解题
方法一:Map排序
利用 时间-价格Map(将时间从小到大排序),可以获得当前时刻的价格(current)
利用 价格-数量Map(将价格从小到大排序), 可以获得最小价格和最大价格(minimum和maximum)
class StockPrice { public: map<int,int> timePriceMap;//时间-价格(按时间从小到大排) map<int,int> priceNumMap;//价格-数量(按价格从小到大排) StockPrice() { } void update(int timestamp, int price) { if(priceNumMap.count(timePriceMap[timestamp])){//如果被覆盖的股票价格 在 价格-数量Map 上存在,就让它数量-1 priceNumMap[timePriceMap[timestamp]]--; if(priceNumMap[timePriceMap[timestamp]]==0) priceNumMap.erase(timePriceMap[timestamp]); } timePriceMap[timestamp]=price;//更新时间-价格 priceNumMap[price]++;//价格-数量+1 } int current() { return timePriceMap.rbegin()->second;//当前的价格 } int maximum() { return priceNumMap.rbegin()->first;//最大的价格 } int minimum() { return priceNumMap.begin()->first;//最小的价格 } };