【bzoj1367】[Baltic2004]sequence

 1 #include<bits/stdc++.h>
 2 #define inf 1000000000
 3 #define ll long long
 4 #define N 1000005
 5 using namespace std;
 6 int read(){
 7   int x=0,f=1;char ch=getchar();
 8   while(ch<‘0‘||ch>‘9‘){if(ch==‘-‘)f=-1;ch=getchar();}
 9   while(ch>=‘0‘&&ch<=‘9‘){x=x*10+ch-‘0‘;ch=getchar();}
10   return x*f;
11 }
12 int n,tot,rt[N],v[N],dep[N],ls[N],rs[N],l[N],r[N],cnt[N],num[N];
13 int merge(int x,int y){
14   if(!x||!y)return x+y;
15   if(v[x]<v[y])swap(x,y);
16   rs[x]=merge(rs[x],y);
17   if(dep[ls[x]]<dep[rs[x]])swap(ls[x],rs[x]);
18   dep[x]=dep[rs[x]]+1;
19   return x;
20 }
21 int main(){
22   n=read();
23   for(int i=1;i<=n;i++)v[i]=read()-i;
24   for(int i=1;i<=n;i++){
25     ++tot;
26     rt[tot]=i;cnt[tot]=1;num[tot]=1;
27     l[tot]=r[tot]=i;
28     while(tot>1&&v[rt[tot]]<v[rt[tot-1]]){
29       --tot;
30       rt[tot]=merge(rt[tot],rt[tot+1]);
31       num[tot]+=num[tot+1];cnt[tot]+=cnt[tot+1];r[tot]=r[tot+1];
32       for(;cnt[tot]*2>num[tot]+1;cnt[tot]--)rt[tot]=merge(ls[rt[tot]],rs[rt[tot]]);
33     }
34   }
35   ll ans=0;
36   for(int i=1;i<=tot;i++)
37     for(int j=l[i],w=v[rt[i]];j<=r[i];j++)
38       ans+=abs(v[j]-w);
39   printf("%lld\n",ans);
40   return 0;
41 }