标签:bzoj oi stl
1058: [ZJOI2007]报表统计
Time Limit: 15 Sec Memory Limit: 162 MB
Submit: 2013 Solved: 708
[Submit][Status]
Description
小Q的妈妈是一个出纳,经常需要做一些统计报表的工作。今天是妈妈的生日,小Q希望可以帮妈妈分担一些工作,作为她的生日礼物之一。经过仔细观察,小Q发现统计一张报表实际上是维护一个可能为负数的整数数列,并且进行一些查询操作。在最开始的时候,有一个长度为N的整数序列,并且有以下三种操作: INSERT i k 在原数列的第i个元素后面添加一个新元素k;
如果原数列的第i个元素已经添加了若干元素,则添加在这些元素的最后(见下面的例子) MIN_GAP 查询相邻两个元素的之间差值(绝对值)的最小值 MIN_SORT_GAP 查询所有元素中最接近的两个元素的差值(绝对值) 例如一开始的序列为 5 3 1 执行操作INSERT 2 9将得到: 5 3 9 1 此时MIN_GAP为2,MIN_SORT_GAP为2。 再执行操作INSERT 2 6将得到: 5 3 9 6 1 注意这个时候原序列的第2个元素后面已经添加了一个9,此时添加的6应加在9的后面。这个时候MIN_GAP为2,MIN_SORT_GAP为1。于是小Q写了一个程序,使得程序可以自动完成这些操作,但是他发现对于一些大的报表他的程序运行得很慢,你能帮助他改进程序么?
Input
第一行包含两个整数N,M,分别表示原数列的长度以及操作的次数。第二行为N个整数,为初始序列。接下来的M行每行一个操作,即“INSERT i k”,“MIN_GAP”,“MIN_SORT_GAP”中的一种(无多余空格或者空行)。
Output
对于每一个“MIN_GAP”和“MIN_SORT_GAP”命令,输出一行答案即可。
Sample Input
3 5
5 3 1
INSERT 2 9
MIN_SORT_GAP
INSERT 2 6
MIN_GAP
MIN_SORT_GAP
Sample Output
2
2
1
HINT
对于100%的数据,N , M ≤500000 对于所有的数据,序列内的整数不超过5*10^8。
平衡树的题,我用STL水过去的。。
首先说STL的做法:
st[i],ed[i]是类似于链表的数组,表示i这一位第一个插入得数和最后一个插入的数。
b是multiset类型,记录的是当前所有的数字。
a是multiset类型(因为有删除操作),记录的是所有相邻数的差值。
q是优先队列,记录的是与每一个点权值相邻的数的差值。
对于Insert(p,x),首先要在a中删除p与p+1原本相邻点的差值,加入目前差值;
并在q中加入与x权值相邻的数的差值;
最后修改ed数组。
查询直接输出第一个即可。
#include <iostream>
#include <cstdio>
#include <map>
#include <algorithm>
#include <cstring>
#include <cstdlib>
#include <queue>
#include <set>
#define M 500005
using namespace std;
char str[50];
int st[M],ed[M],n,m;
multiset<int> a,b;
map<int,int> mp;
priority_queue<int,vector<int>,greater<int> > q;
void read(int &tmp)
{
tmp=0;
char ch=getchar();
int fu=1;
for (;ch<'0'||ch>'9';ch=getchar())
if (ch=='-') fu=-1;
for (;ch>='0'&&ch<='9';ch=getchar())
tmp=tmp*10+ch-'0';
tmp*=fu;
}
void Insert(int x)
{
mp[x]++;
if (mp[x]==1) a.insert(x);
}
void Push(int x)
{
int l=*--b.lower_bound(x),r=*b.lower_bound(x);
q.push(min(x-l,r-x));
b.insert(x);
}
int main()
{
read(n),read(m);
int x;
int inf=1000000000;
b.insert(inf),b.insert(-inf);
for (int i=1;i<=n;i++)
read(x),Push(x),st[i]=ed[i]=x;
for (int i=2;i<=n;i++)
{
int now=abs(st[i]-st[i-1]);
Insert(now);
}
for (int i=1;i<=m;i++)
{
scanf("%s",str);
if (str[0]=='I')
{
int p,x;
read(p),read(x);
if (p!=n)
{
int now=abs(ed[p]-st[p+1]);
mp[now]--;
if (!mp[now]) a.erase(now);
Insert(abs(x-st[p+1]));
}
Insert(abs(ed[p]-x));
ed[p]=x;
Push(x);
}
else if (str[4]=='S') printf("%d\n",q.top());
else printf("%d\n",*a.begin());
}
return 0;
}
接下来说一下标准做法:
就是把STL的部分换成splay来维护,每个点记录与他相邻点的权值,以及以他为根的子树的最小值;
因为有两种询问,所以要开两种splay(以位置为关键字,以权值为关键字)
【BZOJ 1058】 [ZJOI2007]报表统计
标签:bzoj oi stl
原文地址:http://blog.csdn.net/regina8023/article/details/43369819