class StockPrice {
int maxtimestamp; //全局 -- 用于返回当前时间戳对应的价格
HashMap<Integer,Integer>timepricestamp;
TreeMap<Integer,Integer>prices;
public StockPrice() {
int maxtimestamp = 0;
timepricestamp = new HashMap<Integer,Integer>();
prices = new TreeMap<Integer,Integer>();
}
//举例子1,10 更新 1,3就理解了
public void update(int timestamp, int price) {
maxtimestamp = Math.max(maxtimestamp,timestamp);
int prevPrice = timepricestamp.getOrDefault(timestamp,0);//判断此时间戳是否出现过价格,就比如之前出现过1,10 ,之后又出现1,3 那么1,10就是错误的
timepricestamp.put(timestamp,price);
if(prevPrice > 0){//表示此时间戳之前出现过价格,之前的就视为错的,要更新,timepricestamp可以直接覆盖,但是由于所有价格是一个有序集合,TreeMap,就要将错误的减掉,减去出现次数,如果次数减到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 timepricestamp.get(maxtimestamp);
}
public int maximum() {
return prices.lastKey();
}
public int minimum() {
return prices.firstKey();
}
}
哈希表 + TreeMap维护有序列表(2034. 股票价格波动)
最新推荐文章于 2024-06-13 08:37:22 发布