题目描述:
给你一支股票价格的数据流。数据流中每一条记录包含一个 时间戳 和该时间点股票对应的 价格 。
不巧的是,由于股票市场内在的波动性,股票价格记录可能不是按时间顺序到来的。某些情况下,有的记录可能是错的。如果两个有相同时间戳的记录出现在数据流中,前一条记录视为错误记录,后出现的记录 更正 前一条错误的记录。
请你设计一个算法,实现:
更新 股票在某一时间戳的股票价格,如果有之前同一时间戳的价格,这一操作将 更正 之前的错误价格。
找到当前记录里 最新股票价格 。最新股票价格 定义为时间戳最晚的股票价格。
找到当前记录里股票的 最高价格 。
找到当前记录里股票的 最低价格 。
请你实现 StockPrice 类:
StockPrice() 初始化对象,当前无股票价格记录。
void update(int timestamp, int price) 在时间点 timestamp 更新股票价格为 price 。
int current() 返回股票 最新价格 。
int maximum() 返回股票 最高价格 。
int minimum() 返回股票 最低价格 。
方法一:
class StockPrice {
public:
unordered_map<int, int> maps; //存储一个时间戳对应的价格
int latesttime = INT_MIN; //记录最新时间
//优先队列的第一个元素为价格,第二个元素为时间
//分别按价格降序排列和升序排列
priority_queue<pair<int, int>> maxpriceque;
priority_queue<pair<int, int>, vector<pair<int, int>>, greater<pair<int, int>>> minpriceque;
StockPrice() {
}
void update(int timestamp, int price) {
latesttime = max(latesttime, timestamp);
maps[timestamp] = price;
maxpriceque.push(pair<int, int>(price, timestamp));
minpriceque.push(pair<int, int>(price, timestamp));
}
int current() {
return maps[latesttime];
}
int maximum() {
//前者是maxpriceque的首元素对应的时间下,maps中的价格(maps中的价格一定是最后更新的)
//后者是maxpriceque的首元素对应的价格
//若两者不相等,说明maxpriceque的首元素是个之前的错误记录,应该删除掉
while (maps[maxpriceque.top().second] != maxpriceque.top().first)
{
maxpriceque.pop();
}
return maxpriceque.top().first;
}
int minimum() {
//前者是minpriceque的首元素对应的时间下,maps中的价格(maps中的价格一定是最后更新的)
//后者是minpriceque的首元素对应的价格
//若两者不相等,说明minpriceque的首元素是个之前的错误记录,应该删除掉
while (maps[minpriceque.top().second] != minpriceque.top().first)
{
minpriceque.pop();
}
return minpriceque.top().first;
}
};
又是看题解的一天呢~
思路是对的,就是利用哈希表和优先队列实现,但是一直不知道在更新某个时间戳的价格时,如何在优先队列中删除原有的记录,看了别人的方法,就将优先队列中的元素使用pair,且pair的第一个元素为价格,第二个元素为时间即可!
感觉这样实在是太妙了,在后续求最高、最低价格时,队列中存储的每个价格都有对应的时间,由于maps中存储的一定是正确的价格,所以只要判断优先队列中的时间所对应的价格是否是正确的就可以确定是不是一个可以删除的错误记录了!