Leetcode 2034 股票价格波动

给你一支股票价格的数据流。数据流中每一条记录包含一个 时间戳 和该时间点股票对应的 价格 。

不巧的是，由于股票市场内在的波动性，股票价格记录可能不是按时间顺序到来的。某些情况下，有的记录可能是错的。如果两个有相同时间戳的记录出现在数据流中，前一条记录视为错误记录，后出现的记录 更正 前一条错误的记录。

请你设计一个算法，实现：

更新 股票在某一时间戳的股票价格，如果有之前同一时间戳的价格，这一操作将 更正 之前的错误价格。
找到当前记录里 最新股票价格 。最新股票价格 定义为时间戳最晚的股票价格。
找到当前记录里股票的 最高价格 。
找到当前记录里股票的 最低价格 。
请你实现 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();
    }
}