【每日一题Day341】LC2034股票价格波动

本文链接：https://blog.csdn.net/Tikitian/article/details/133684280

股票价格波动【LC2034】

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

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

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

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

请你实现 StockPrice 类：

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

放假回家电脑坏了国庆休息的好舒服继续打卡

思路
- 使用哈希表存储每个时间点对应的价格，并使用小顶堆和大顶堆记录二元组 $[时间，价格]$ ，以价格进行排序，同时记录最大时间点maxTime
- update：将数据记录在哈希表、小顶堆、大顶堆，并判断是否需要更新最大时间点maxTime
- current：从哈希表中获得maxTime对应的价格
- maximum/minimum：如果一个时间点的价格与哈希表中不一致，那么不是有效数据，将大顶堆/小顶堆中的无效数据移除后，返回堆顶元素

实现

class StockPrice {
    // 如果发生了价格更正，那么会影响最高价格和最低价格
    // 使用小顶堆、大顶堆存放二元组[时间，价格]，哈希表记录每个时间对应的最新价格 堆顶元素时间对应的价格必须与哈希表一致
    // 记录最大时间
    PriorityQueue<int[]> mxPq;
    PriorityQueue<int[]> mnPq;
    int mxTime;
    Map<Integer,Integer> map;

    public StockPrice() {
        this.mxPq = new PriorityQueue<>((o1, o2) -> o2[1] - o1[1]);
        this.mnPq = new PriorityQueue<>((o1, o2) -> o1[1] - o2[1]);
        mxTime = -1;
        map = new HashMap<>();
    }
    
    public void update(int timestamp, int price) {
        map.put(timestamp, price);
        mxTime = Math.max(mxTime, timestamp);
        mxPq.add(new int[]{timestamp, price});
        mnPq.add(new int[]{timestamp, price});
    }
    
    public int current() {
        return map.get(mxTime);
    }
    
    public int maximum() {
        // 懒删除
        while (mxPq.peek()[1] != map.get(mxPq.peek()[0])){
            mxPq.poll();
        }
        return mxPq.peek()[1];
    }
    
    public int minimum() {
        // 懒删除
        while (mnPq.peek()[1] != map.get(mnPq.peek()[0])){
            mnPq.poll();
        }
        return mnPq.peek()[1];
    }
}

/**
 * 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();
 */