[hihoCoder] #1306 : 股票价格 解题报告 - Jonah的专栏 - 博客频道 - CSDN.NET
Read full article from [hihoCoder] #1306 : 股票价格 解题报告 - Jonah的专栏 - 博客频道 - CSDN.NET
小Hi最近在分析一支股票的价格走势，他需要一个程序来辅助分析。这个程序会接收3种消息（指令）：
价格信息，格式是P timestamp price：表示这支股票在 timestamp 时刻价格是 price。
删除价格指令，格式是R timestamp：随着时间推移，小Hi会积累越来越多的价格数据。一些老旧的数据会变得不重要。这个指定会删除 timestamp 以前（包括 timestamp 时刻）的价格数据。
价格查询指令，格式是Q：小Hi希望程序返回这只股票最高、最低和最近的价格。注意已经被删除的价格不应该被统计。
给定一个包含以上3种信息（指令）的序列，你能否帮助小Hi完成这个程序呢？
输入
第1行包含一个整数 N (1 ≤ N ≤ 500000)，表示消息（指令）序列的长度。
第2 - N+1行，每行包含一条消息或指令。
输入保证价格信息是按照 timestamp 升序排列的，并且出现的 timestamp 和价格小于100000000。
输出
对于输入中每一条价格查询指令，输出当时最高、最低和最近的价格。
- 样例输入
10 P 1 77 P 2 73 P 5 70 P 7 74 Q R 4 Q P 8 78 R 5 Q
- 样例输出
77 70 74 74 70 74 78 74 78
思路: 因为涉及到删除以前某个时间以前的信息, 并且还要统计最高和最低的股票价格, 因此可以用一个map来保存以时间为key的股票价格, 然后以二叉搜索数set保存以股票价格和时间的对. 这样做的好处是map是本身有序的, 这样当我们要删除信息的时候最前面的就是最早的股票信息, 然后我们从map中拿到要删除的股票信息到set中查找要删除的股票, 因为是二叉搜索树, 所有其查找和删除时间复杂度都是O(log n), 还是可以接受的.
int main()
{
int N, time, price, curTime;
char ch;
set<pair<int, int> > st;
map<int, int> hash;
cin>>N;
for(int i =0; i< N; i++)
{
cin>>ch;
if(ch == 'P')
{
cin >> time >> price;
st.insert(make_pair(price, time));
hash[time] = price;
curTime = time;
}
else if(ch == 'Q')
cout << st.rbegin()->first << " " << st.begin()->first << " " << hash[curTime] << endl;
else if(ch == 'R')
{
int dtime;
cin >> dtime;
while(hash.begin()->first<= dtime)
{
st.erase(make_pair(hash.begin()->second, hash.begin()->first));
hash.erase(hash.begin());
}
}
}
return 0;
}
Read full article from [hihoCoder] #1306 : 股票价格 解题报告 - Jonah的专栏 - 博客频道 - CSDN.NET
