Leetcode.2034 股票价格波动

题目链接

Leetcode.2034 股票价格波动 rating : 1832

题目描述

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

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

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

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

请你实现 $StockPrice$ 类：

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

示例 1：

输入：
[“StockPrice”, “update”, “update”, “current”, “maximum”, “update”, “maximum”, “update”, “minimum”]
[[], [1, 10], [2, 5], [], [], [1, 3], [], [4, 2], []]
输出：
[null, null, null, 5, 10, null, 5, null, 2]

解释： StockPrice stockPrice = new StockPrice(); stockPrice.update(1,
10); // 时间戳为 [1] ，对应的股票价格为 [10] 。 stockPrice.update(2, 5); // 时间戳为
[1,2] ，对应的股票价格为 [10,5] 。 stockPrice.current(); // 返回 5 ，最新时间戳为 2
，对应价格为 5 。 stockPrice.maximum(); // 返回 10 ，最高价格的时间戳为 1 ，价格为 10 。
stockPrice.update(1, 3); // 之前时间戳为 1 的价格错误，价格更新为 3 。
// 时间戳为 [1,2] ，对应股票价格为 [3,5] 。 stockPrice.maximum(); // 返回 5 ，更正后最高价格为 5 。 stockPrice.update(4,
2); // 时间戳为 [1,2,4] ，对应价格为 [3,5,2] 。 stockPrice.minimum(); // 返回
2 ，最低价格时间戳为 4 ，价格为 2 。

提示：

• $1\le timestamp,price\le 10^9$
• update，current，maximum 和 minimum 总 调用次数不超过 $10^5$ 。
• current，maximum 和 minimum 被调用时，update 操作 至少 已经被调用过 一次 。

解法：哈希表 + 模拟

用一个哈希表 $m$ 来记录，股票时间戳 和 价格的关系，即 [ 时间戳 ， 价格 ] 。

用另外一个哈希表 $p$ 来记录，同一个价格下的不同时间戳，即 [ 价格1 : [时间戳1 , 时间戳2 , 时间戳3] ]。

$m$ 和 $p$ 都是 自动按照键值从小到大排序的。

这样我们就能快速获取 当前股票的价格，最高价格，最低价格。

我们只需要维护这两个哈希表即可。

时间复杂度：current，maximum ， minimum 和 update 操作都是 $O(1)$

C++代码：

class StockPrice {
public:
    map<int,int> m;
    map<int,unordered_set<int>> p;

    StockPrice() {

    }
    
    void update(int timestamp, int price) {
        if(!m.count(timestamp)){
            m[timestamp] = price;
            p[price].insert(timestamp);
        }
        else{
            int old_price = m[timestamp];
            p[old_price].erase(timestamp);

            if(p[old_price].size() == 0) p.erase(old_price);

            m[timestamp] = price;
            p[price].insert(timestamp);
        }
    }
    
    int current() {
        auto it = m.end();
        --it;

        return it->second;
    }
    
    int maximum() {
        auto it = p.end();
        --it;
        return it->first;
    }
    
    int minimum() {
        auto it = p.begin();
        return it->first;
    }
};

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