题目描述
给你一支股票价格的数据流。数据流中每一条记录包含一个 时间戳 和该时间点股票对应的 价格 。
不巧的是,由于股票市场内在的波动性,股票价格记录可能不是按时间顺序到来的。某些情况下,有的记录可能是错的。如果两个有相同时间戳的记录出现在数据流中,前一条记录视为错误记录,后出现的记录 更正 前一条错误的记录。
请你设计一个算法,实现:
更新 股票在某一时间戳的股票价格,如果有之前同一时间戳的价格,这一操作将 更正 之前的错误价格。
找到当前记录里 最新股票价格 。最新股票价格 定义为时间戳最晚的股票价格。
找到当前记录里股票的 最高价格 。
找到当前记录里股票的 最低价格 。
请你实现 StockPrice 类:
StockPrice() 初始化对象,当前无股票价格记录。
void update(int timestamp, int price) 在时间点 timestamp 更新股票价格为 price 。
int current() 返回股票 最新价格 。
int maximum() 返回股票 最高价格 。
int minimum() 返回股票 最低价格 。
做题情况
- 做出来且思路与标答一致 ☑
- 做出来但思路较为复杂
- 有思路,但时间复杂度较高无法通过
- 没有思路
自己的想法:unordered_map+multiset
首先返回股票的最新价格是很好求的,我们只需要在每次更新的时候判断这个时间是否是晚于当前最晚时间即可。如果晚于,则进行覆盖。另外需要注意的时,如果等于也需要进行修改,因为会存在当前时间股票价格更新的情况。
其次是股票更新以及求股票最大值最小值这两个问题。首先观察问题规模,最大可以用O(nlogn)的时间复杂度,由于需要进行n次操作,因此每次插入修改的时间复杂度为O(logn)。
对于股票更新,首先想到的就是unordered_map,修改,插入均为O(1)时间复杂度。
对于最大值最小值,在O(logn)时间复杂度的即为树结构。首先想到的是set。但由于不同时间的股票价格可能相同,而set中不允许相同的元素出现,因此我们使用multiset来进行储存所有的股票价格。
有一点需要注意:当我们删除multiset中的元素时,可以先找到,再删除,如果直接调用erase()的话,会把相同值全部删除。
还有一个细节,就是要得到multiset的最后一个元素时,自己是用auto it=mu_set.end(); it--; return *it;
来表示的,但可以直接用return mu_set.rbegin();
来表示,可以学习一下。
标答:
同自己的想法
实际代码
class StockPrice {
public:
multiset<int> mu_set;
unordered_map<int,int> un_map;
int current_time;
int current_res;
StockPrice()
{
mu_set.clear();
un_map.clear();
current_time=0;
current_res=-1;
}
void update(int timestamp, int price)
{
if (timestamp>=current_time)
{
current_time=timestamp;
current_res=price;
}
if (un_map[timestamp]!=0)
{
auto it=mu_set.find(un_map[timestamp]);
mu_set.erase(it);
mu_set.insert(price);
un_map[timestamp]=price;
}
else
{
mu_set.insert(price);
un_map[timestamp]=price;
}
}
int current()
{
return current_res;
}
int maximum()
{
auto it=mu_set.end();
it--;
return *it;
}
int minimum()
{
auto it=mu_set.begin();
return *it;
}
};
总结
填充类的题目考查的都是对数据结构的熟练使用。知道最大允许的时间,要进行什么样的操作,就可以找到对应的数据结构来进行使用。这其实是一个积累的过程。