题意:
给一个序列 Ti T i ,求构造一个递增序列 zi z i ,使 ∑ni|Ti−zi| ∑ i n | T i − z i | 最小。
题解:
感觉上挺套路的题吧虽然我不会做
假如
T
T
是一个上升序列,那么显然。
假如
T
T
是一个下降序列,一个常用的结论,当即中位时最优。
这就告诉我们,当
T
T
<script type="math/tex" id="MathJax-Element-471">T</script>中连续的一段都是单调下降时,可以将他们合并,用中位数作为新权,一直不断合并直到合并成一个单调上升的序列。
中位数用左偏树维护即可。
code:
#include<cmath>
#include<cstdio>
#include<cstdlib>
#include<cstring>
#include<iostream>
#define LL long long
using namespace std;
int n;
struct trnode{
int tot,l[1000010],r[1000010],d[1000010],size[1000010],c[1000010];
int merge(int x,int y)
{
if(!x||!y) return x+y;
if(c[x]<c[y]) swap(x,y);
r[x]=merge(r[x],y);
size[x]=size[l[x]]+size[r[x]]+1;
if(d[r[x]]>d[l[x]]) swap(l[x],r[x]);
d[x]=d[r[x]]+1;
return x;
}
int top(int x) {return c[x];}
int Size(int x) {return size[x];}
void pop(int &x) {x=merge(l[x],r[x]);}
int push(int _c)
{
int x=++tot;
c[x]=_c;size[x]=1;
d[x]=l[x]=r[x]=0;
return x;
}
}q;
int a[1000010];
int num=0,root[1000010],tot[1000010],L[1000010],R[1000010];
int main()
{
scanf("%d",&n);
for(int i=1;i<=n;i++) scanf("%d",&a[i]),a[i]-=i;
for(int i=1;i<=n;i++)
{
num++;root[num]=q.push(a[i]);
tot[num]=1;L[num]=R[num]=i;
while(num>1&&q.top(root[num-1])>q.top(root[num]))
{
num--;
root[num]=q.merge(root[num],root[num+1]);
tot[num]+=tot[num+1];R[num]=R[num+1];
while(q.Size(root[num])*2>tot[num]+1) q.pop(root[num]);
}
}
LL ans=0;
for(int i=1;i<=num;i++)
for(int j=L[i],t=q.top(root[i]);j<=R[i];j++)
ans+=abs(a[j]-t);
printf("%lld",ans);
}