题目1 : 股票价格
-
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
描述
小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。
输出
对于输入中每一条价格查询指令,输出当时最高、最低和最近的价格。
思路:这道题线段树可解。看这道题的第一眼就想到了线段树,综合时间复杂度NlogN,对于50万的数据来说完全够用。使用线段树要解决两个问题,如何构造和如何查询。
构造时可以预先把所有股票价格保存到一个数组中,然后构造一颗线段树,每个节点存两个值,最大与最小。剩下的就是确定查找范围了,可以用两个变量(初始化0),存查找范围。第一个在遇到删除操作时向后移,移动总时间复杂度为N。第二个变量在遇到添加操作时向后移一位。这样就可以确定查找范围了,故而线段树可解。
代码如下:
#include <iostream>
#include <cstdio>
#include <cstring>;
int a[2000002],b[2000002],c[500005],d[500005];
struct aaa
{
char a;
int b,c;
}aa[500005];
void gouzao(int v,int x,int y)
{
if(x==y)
{
a[v]=c[x];
b[v]=c[x];
return;
}
int mid=(x+y)/2;
gouzao(2*v,x,mid);
gouzao(2*v+1,mid+1,y);
a[v]=std::max(a[2*v],a[2*v+1]);
b[v]=std::min(b[2*v],b[2*v+1]);
}
int dchazhao(int v,int x,int y,int l,int r)
{
if(x==l&&y==r)
return a[v];
int mid=(x+y)/2;
if(mid>=r)
return dchazhao(2*v,x,mid,l,r);
if(mid<l)
return dchazhao(2*v+1,mid+1,y,l,r);
return std::max(dchazhao(2*v,x,mid,l,mid),dchazhao(2*v+1,mid+1,y,mid+1,r));
}
int xchazhao(int v,int x,int y,int l,int r)
{
if(x==l&&y==r)
return b[v];
int mid=(x+y)/2;
if(mid>=r)
return xchazhao(2*v,x,mid,l,r);
if(mid<l)
return xchazhao(2*v+1,mid+1,y,l,r);
return std::min(xchazhao(2*v,x,mid,l,mid),xchazhao(2*v+1,mid+1,y,mid+1,r));
}
int main()
{
int n,t=0;
std::cin>>n;
for(int i=0;i<n;i++)
{
std::cin>>aa[i].a;
//std::cout<<"=-=\n";
if(aa[i].a=='P')
{
//std::cout<<"0-0\n";
scanf("%d%d",&aa[i].b,&aa[i].c);
d[t]=aa[i].b;
c[t++]=aa[i].c;
}
else if(aa[i].a=='R')
scanf("%d",&aa[i].b);
}
gouzao(1,0,t-1);
int z=0,x=0,y=-1;
for(int i=0;i<n;i++)
{
if(aa[i].a=='P')
{
y++;
z=aa[i].c;
}
else if(aa[i].a=='R')
{
int j;
for(j=x;j<=y;j++)
{
if(d[j]>aa[i].b)
{
x=j;
break;
}
}
if(j==y)
x=y+1;
}
else
{
//std::cout<<x<<" "<<y<<std::endl;
int v=dchazhao(1,0,t-1,x,y),v1=xchazhao(1,0,t-1,x,y);
printf("%d %d %d\n",v,v1,z);
}
}
}