leetcode5896. 股票价格波动 (hash表 multiset 设计类)

链接:https://leetcode-cn.com/problems/stock-price-fluctuation/

题目

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

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

请你设计一个算法,实现:

更新 股票在某一时间戳的股票价格,如果有之前同一时间戳的价格,这一操作将 更正 之前的错误价格。
找到当前记录里 最新股票价格 。最新股票价格 定义为时间戳最晚的股票价格。
找到当前记录里股票的 最高价格 。
找到当前记录里股票的 最低价格 。
请你实现 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 <= timestamp, price <= 109
update,current,maximum 和 minimum 总 调用次数不超过 105 。
current,maximum 和 minimum 被调用时,update 操作 至少 已经被调用过 一次 。

思路

周赛第三题
在更新股票价格情况下 维护最大最小值情况下 由于对stl中容器使用不太熟悉
在每次更新错误价格的情况下 对存储数组排序 取最大最小值
很不幸超时

class StockPrice {
public:
    StockPrice() {

    }
    
    void update(int timestamp, int price) {
        int ischange=0;
        int temp=0;
        if(mp.find(timestamp)!=mp.end())
        {
            ischange=1;
            temp=mp[timestamp];
        }
        mp[timestamp]=price;
        if(!ischange){
            if(price>maxnum)
                maxnum=price;
            if(price<minnum)
                minnum=price;
        }
        else{
            if(minnum==temp||maxnum==temp)
            {
                maxnum=INT_MIN;
                minnum=INT_MAX;
                for(auto it=mp.begin();it!=mp.end();it++)
                {
                    int t=it->second;
                    if(t>maxnum)
                        maxnum=t;
                    if(t<minnum)
                        minnum=t;
                }
            }else{
            if(price>maxnum)
                maxnum=price;
            if(price<minnum)
                minnum=price;
            }
        }
        if(timestamp>last)
            last=timestamp;
        
    }
    
    int current() {
        return mp[last];
    }
    
    int maximum() {
        return maxnum;
    }
    
    int minimum() {
        return minnum;

    }
private:
    int maxnum=INT_MIN;
    int minnum=INT_MAX;
    unordered_map<int,int>mp;
    int last=0;
    
    
};

实际上维护一个有序容器 只需用到stl中的multiset 自动排序 并且允许重复
在更新错误数据时 先将multiset中原数据时间戳对应的价格删去 注意不能直接erase目标值,会导致所有与目标值相等的元素都被删除 应该选择删除multiset中第一次与目标值相等的迭代器指向的元素 即ms.erase(ms.find(mp[timestamp]));

class StockPrice {
public:
    StockPrice() {
    }
    
    void update(int timestamp, int price) {
        if(mp.find(timestamp)!=mp.end())
        {
            ms.erase(ms.find(mp[timestamp]));
        }
        mp[timestamp]=price;
        ms.insert(price);
        if(timestamp>last)
            last=timestamp;
    }
    
    int current() {
        return mp[last];
    }
    
    int maximum() {
        return *ms.rbegin();
    }
    
    int minimum() {
        return *ms.begin();
    }
private:
    unordered_map<int,int>mp;
    multiset<int>ms;
    int last=0;
};
posted @ 2021-10-11 16:35  kitamu  阅读(170)  评论(0)    收藏  举报
Live2D