bzoj 1367: [Baltic2004]sequence（中位数+可并堆）

1367: [Baltic2004]sequence

Time Limit: 20 Sec   Memory Limit: 64 MB
Submit: 935   Solved: 351
[ Submit][ Status][ Discuss]

Description

Input

Output

一个整数R

Sample Input

7
9
4
8
20
14
15
18

Sample Output

13

HINT

所求的Z序列为6,7,8,13,14,15,18.
R=13

Source

[ Submit][ Status][ Discuss]

题解：中位数+可并堆

分情况讨论一下：

一个区间[l, r]，递增的，则对于此区间最优解为z[i] = t[i]；

一个区间[l,r] ，递减的，则z[l] = z[l + 1] = ... = z[r] = 这段数的中位数，不妨叫做w。（中位数为第(r - l + 1) / 2大的数），为什么是中位数呢，因为区间是递减的，我们要把他变成不下降的，最小的改动情况就是把他改成一个每个数都相同的序列，那么这就转化成了中位数问题，因为所有点到中位数的距离是最小的。但是我们要求的是上升序列而不是不下降序列，所以需要将将v[i]处理成v[i]-i。

那么这样就可以做了，对于递增的区间，我们可以把区间中的每一个数看成一个单独的堆，那么这个堆维护的中位数就是他本身。如果要加入一个数，我们可以把这个点先看成一个单独的堆，如果v[root[now]]<v[root[now]]也就是当前这个点，比他的前一个区间的中位数要小，但是我们要求他递增，那么这个问题又转换成了“一个区间[l,r] ，递减的，则z[l] = z[l + 1] = ... = z[r] = 这段数的中位数”这个问题，我们可以用可并堆来将两个合并到一起。

#include<iostream>
#include<cstdio>
#include<cstring>
#include<cmath>
#include<algorithm>
#define N 1000003 
using namespace std;
int n,m;
int v[N],l[N],r[N],R[N],L[N],size[N],tot[N],root[N],d[N];
int merge(int x,int y)
{
	if (!x) return y;
	if (!y) return x;
	if (v[x]<v[y]) swap(x,y);
	R[x]=merge(R[x],y);
	size[x]=size[L[x]]+size[R[x]]+1;
	if (d[L[x]]<=d[R[x]])  swap(L[x],R[x]);
	d[x]=d[R[x]]+1;
	return x;
}
int main()
{
	scanf("%d",&n);
	for (int i=1;i<=n;i++) scanf("%d",&v[i]),v[i]=v[i]-i;
    int now=0;
    for (int i=1;i<=n;i++)
     {
     	now++;
     	root[now]=i; 
        l[now]=r[now]=i; //所控制区间的左右端点
	    tot[now]=1;//当前堆维护的中位数控制的数的个数
        size[root[now]]=1;//堆中元素的个数
        while (now>1&&v[root[now-1]]>v[root[now]])
         {
         	now--;
         	r[now]=r[now+1]; tot[now]+=tot[now+1];
         	root[now]=merge(root[now],root[now+1]);
         	while (size[root[now]]*2>tot[now]+1)//保证堆顶元素是当前区间的中位数
         	 root[now]=merge(L[root[now]],R[root[now]]);
         }
     }
    long long ans=0;
    for (int i=1;i<=now;i++)
     for (int j=l[i];j<=r[i];j++) 
      ans+=(long long)abs(v[j]-v[root[i]]);
    printf("%I64d\n",ans);
}