[hihoCoder] #1306 : 股票价格 解题报告

题目链接: http://hihocoder.com/problemset/problem/1306
时间限制:20000ms
单点时限:2000ms
内存限制:256MB

描述

小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), 还是可以接受的.

代码如下:

/*************************************************************************
> File Name: stock.cpp
> Author: Maoting Ren
> Mail: mren@g.clemson.edu
> Created Time: Sun 22 May 2016 01:28:27 AM EDT
************************************************************************/

#include<iostream>
#include<set>
#include<map>
using namespace std;

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;
}