Leetcode2034 股票价格波动

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

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

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

更新 股票在某一时间戳的股票价格，如果有之前同一时间戳的价格，这一操作将 更正 之前的错误价格。
找到当前记录里 最新股票价格 。最新股票价格 定义为时间戳最晚的股票价格。
找到当前记录里股票的 最高价格 。
找到当前记录里股票的 最低价格 。
请你实现 StockPrice 类：

StockPrice() 初始化对象，当前无股票价格记录。
void update(int timestamp, int price) 在时间点 timestamp 更新股票价格为 price 。
int current() 返回股票 最新价格 。
int maximum() 返回股票 最高价格 。
int minimum() 返回股票 最低价格 。

思路：
记录每个时间点对应的正确股价，记录每个股价对应的时间集合（为方便排序，运用treemap）

class StockPrice {

     private Map<Integer, Integer> timePriceMap;

    private Integer current;
    private TreeMap<Integer, Set<Integer>> priceTimesMap;

    public StockPrice() {
         current = 0;
        this.timePriceMap = new HashMap<>();
        this.priceTimesMap = new TreeMap<>();
    }

    public void update(int timestamp, int price) {
        if(timestamp > current)
        this.current = timestamp;

        if (timePriceMap.containsKey(timestamp)) {  // 之前出现过
            int oldPrice = timePriceMap.get(timestamp);
            Set<Integer> times = priceTimesMap.get(oldPrice);
            times.remove((Integer) timestamp);
            if(times.size() == 0) {
                priceTimesMap.remove(oldPrice);
            }
        }
        timePriceMap.put(timestamp, price);
        Set<Integer> times = priceTimesMap.getOrDefault(price, new HashSet<>());
        times.add(timestamp);
        priceTimesMap.put(price, times);
    }

    public int current() {
        return this.timePriceMap.get(current);
    }

    public int maximum() {
        return priceTimesMap.lastKey();
    }

    public int minimum() {
        return priceTimesMap.firstKey();
    }
}