【堆+平衡树】BZOJ1058(ZJOI2007)[报表统计]题解

题目概述

维护一个可能为负数的整数数列。在最开始的时候，有一个长度为 $n$ 的整数序列，并且有以下三种操作：

$INSERT\ i\ k$ ：在原数列的第 $i$ 个元素后面添加一个新元素 $k$ ； 如果原数列的第 $i$ 个元素已经添加了若干元素，则添加在这些元素的最后。

$MIN\_GAP$ ：查询相邻两个元素之间的差值（绝对值）的最小值。

$MIN\_SORT\_GAP$ ：查询所有元素中最接近的两个元素的差值（绝对值） 。

解题报告

可以认为插入操作是挂链的形式。然后用堆维护相邻两个元素的差值的最小值，再用平衡树维护前驱和后继来求最接近的两个元素的差值。STL大法好……

示例程序

#include<cstdio>
#include<cctype>
#include<cstring>
#include<algorithm>
#include<vector>
#include<set>
#define fr first
#define sc second
#define mp make_pair
using namespace std;
typedef pair<int, pair<int,int> > data;
const int maxn=500000,maxm=500000;

int n,m,a[maxn+5],MS;vector<int> v[maxn+5];
set<int> s;set<int>::iterator it;
int si;data Heap[maxn+2*maxm+5];

#define Abs(x) ((x)<0?-(x):(x))
inline bool cmp(data a,data b) {return a.fr>b.fr;}
inline int Pre(int i,int j) {int a=i,b=j-1;if (b<0) a--,b=v[a].size()-1;return Abs(v[i][j]-v[a][b]);}
inline void Push(int i,int j) {Heap[++si]=mp(Pre(i,j),mp(i,j));push_heap(Heap+1,Heap+1+si,cmp);}
inline int Top()
{
    while (Heap[1].fr!=Pre(Heap[1].sc.fr,Heap[1].sc.sc)) pop_heap(Heap+1,Heap+1+si--,cmp);
    return Heap[1].fr;
}
inline void Insert(int i,int x)
{
    it=s.lower_bound(x);if (it!=s.end()) MS=min(MS,*it-x);if (it!=s.begin()) it--,MS=min(MS,x-*it);
    v[i].push_back(x);s.insert(x);Push(i,v[i].size()-1);if (v[i+1].size()) Push(i+1,0);
}
int main()
{
    freopen("program.in","r",stdin);
    freopen("program.out","w",stdout);
    scanf("%d%d",&n,&m);MS=2e9;
    for (int i=1;i<=n;i++) scanf("%d",&a[i]);
    v[1].push_back(a[1]);s.insert(a[1]);
    for (int i=2;i<=n;i++) Insert(i,a[i]);
    for (char s[100];m;m--)
    {
        scanf("%s",s);int x,y;
        switch(s[4])
        {
            case 'R':scanf("%d%d",&x,&y);Insert(x,y);break;
            case 'G':printf("%d\n",Top());break;
            case 'S':printf("%d\n",MS);break;
        }
    }
    return 0;
}