2034.股票价格波动
题目
给你一支股票价格的数据流。数据流中每一条记录包含一个 时间戳 和该时间点股票对应的 价格 。
不巧的是,由于股票市场内在的波动性,股票价格记录可能不是按时间顺序到来的。某些情况下,有的记录可能是错的。如果两个有相同时间戳的记录出现在数据流中,前一条记录视为错误记录,后出现的记录 更正 前一条错误的记录。
请你设计一个算法,实现:
● 更新 股票在某一时间戳的股票价格,如果有之前同一时间戳的价格,这一操作将 更正 之前的错误价格。
● 找到当前记录里 最新股票价格 。最新股票价格 定义为时间戳最晚的股票价格。
● 找到当前记录里股票的 最高价格 。
● 找到当前记录里股票的 最低价格 。
请你实现 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 <= 10⁹
● update,current,maximum 和 minimum 总 调用次数不超过 10⁵。
● current,maximum 和 minimum 被调用时,update 操作 至少 已经被调用过 一次 。
思路
通过哈希表存储时间戳和价格的映射,通过有序集合存储价格和出现次数的映射,用 maxTimestamp 记录最新的时间戳。
通过最新时间戳在哈希表中找到最新的价格,通过有序集合中价格出现的次数来维护价格的存储,若次数为 0 则该价格不存在,如果不为 0 就可以按照价格高低在有序集合中存储并排序,便于返回股票的最高和最低价格。
解答
class StockPrice {
//记录最新的记录对应的时间戳
int maxTimestamp;
//记录时间戳和对应的价格
HashMap<Integer, Integer> timePriceMap;
//有序集合,存储价格和对应的出现次数,用于获取最大值和最小值
TreeMap<Integer, Integer> prices;
public StockPrice() {
maxTimestamp = 0;
timePriceMap = new HashMap<Integer, Integer>();
prices = new TreeMap<Integer, Integer>();
}
public void update(int timestamp, int price) {
//保持记录最新的时间戳
maxTimestamp = Math.max(maxTimestamp, timestamp);
//getOrDefault方法传入键和默认值,如果存在该键,就返回对应的值,不存在则返回默认值
//如果时间价格映射中存在本次传入的时间戳,就将对应的价格传给prevPrice,否则prevPrice为0,表示不存在该键
int prevPrice = timePriceMap.getOrDefault(timestamp, 0);
//无论之前是否存储过timestamp,都直接put,存储过则更新,没存储过则添加
timePriceMap.put(timestamp, price);
//如果prevPrice等于0,说明这个时间戳是第一次出现,不需要删除某个价格的出现次数
//如果prevPrice大于0,说明之前存储过本次时间戳,需要删除一次这个时间戳对应的价格的出现次数
if(prevPrice > 0){
//将prevPrice对应的次数减1
prices.put(prevPrice, prices.get(prevPrice) - 1);
//如果减1后不为0,说明这个价格还有出现次数,在集合中寻找这个价格还是能找得到的
//如果减1后为0,那么就在有序集合中删除这个价格的键值对,也就是这个价格出现的次数为0了,没有这个价格了
if(prices.get(prevPrice) == 0){
prices.remove(prevPrice);
}
}
//不论是否进行价格出现次数减1的操作,都要将price的键值对进行put,并将次数加1
prices.put(price, prices.getOrDefault(price, 0) + 1);
}
public int current() {
//通过最新的时间戳获取最新价格
return timePriceMap.get(maxTimestamp);
}
public int maximum() {
//获取最高价格
return prices.lastKey();
}
public int minimum() {
//获取最低价格
return prices.firstKey();
}
}
/**
* 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();
*/
纯代码
class StockPrice {
int maxTimestamp;
HashMap<Integer, Integer> timePriceMap;
TreeMap<Integer, Integer> prices;
public StockPrice() {
maxTimestamp = 0;
timePriceMap = new HashMap<Integer, Integer>();
prices = new TreeMap<Integer, Integer>();
}
public void update(int timestamp, int price) {
maxTimestamp = Math.max(maxTimestamp, timestamp);
int prevPrice = timePriceMap.getOrDefault(timestamp, 0);
timePriceMap.put(timestamp, price);
if(prevPrice > 0){
prices.put(prevPrice, prices.get(prevPrice) - 1);
if(prices.get(prevPrice) == 0){
prices.remove(prevPrice);
}
}
prices.put(price, prices.getOrDefault(price, 0) + 1);
}
public int current() {
return timePriceMap.get(maxTimestamp);
}
public int maximum() {
return prices.lastKey();
}
public int minimum() {
return prices.firstKey();
}
}
/**
* 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();
*/
如果对您有帮助,请点赞关注支持我,谢谢!❤
如有错误或者不足之处,敬请指正!❤