给你一支股票价格的数据流。数据流中每一条记录包含一个 时间戳 和该时间点股票对应的 价格 。
不巧的是,由于股票市场内在的波动性,股票价格记录可能不是按时间顺序到来的。某些情况下,有的记录可能是错的。如果两个有相同时间戳的记录出现在数据流中,前一条记录视为错误记录,后出现的记录 更正 前一条错误的记录。
请你设计一个算法,实现:
更新 股票在某一时间戳的股票价格,如果有之前同一时间戳的价格,这一操作将 更正 之前的错误价格。
找到当前记录里 最新股票价格 。最新股票价格 定义为时间戳最晚的股票价格。
找到当前记录里股票的 最高价格 。
找到当前记录里股票的 最低价格 。
请你实现 StockPrice 类:
StockPrice() 初始化对象,当前无股票价格记录。
void update(int timestamp, int price) 在时间点 timestamp 更新股票价格为 price 。
int current() 返回股票 最新价格 。
int maximum() 返回股票 最高价格 。
int minimum() 返回股票 最低价格 。
解题思路
哈希表+模拟。这题知道TreeMap的话就很好想,TreeMap是默认将键值升序排列。先用HashMap将时间和价格映射一下,再用TreeMap将价格和数量映射一下,每次更新先判断传入的价格 price
存在否,如果不存在则直接更新俩哈希表;如果存在则先把老价格 oldPrice
拿出来,再更新 oldPrice
对应的数量,再把 price
在俩哈希表中更新。其中最大值是TreeHash的尾键,最小值是其首键。
代码
class StockPrice {
private HashMap<Integer, Integer> time2Price = new HashMap<>();
private TreeMap<Integer, Integer> price2Cnt = new TreeMap<>();
private int nowTime;
public StockPrice() {
}
public void update(int timestamp, int price) {
nowTime = Math.max(nowTime, timestamp);
if (time2Price.containsKey(timestamp)) {
int oldPrice = time2Price.get(timestamp);
int cnt = price2Cnt.get(oldPrice);
if (cnt == 1) price2Cnt.remove(oldPrice);
else price2Cnt.put(oldPrice, cnt - 1);
}
time2Price.put(timestamp, price);
price2Cnt.put(price, price2Cnt.getOrDefault(price, 0) + 1);
}
public int current() {
return time2Price.get(nowTime);
}
public int maximum() {
return price2Cnt.lastKey();
}
public int minimum() {
return price2Cnt.firstKey();
}
}