bzoj1367 [Baltic2004]sequence

左偏树论文模板题,不过想法很好。。
把序列分成m个区间,答案就是m个区间的中位数,中位数的维护,我们每次只维护当前区间前一半的最大堆,然后如果超过一半就弹出,堆顶就是中位数。

#include<cstdio>
#include<algorithm>
#include<cstring>
#include<iostream>
#define fo(i,a,b) for(int i=a;i<=b;i++)
#define fd(i,a,b) for(int i=a;i>=b;i--)
using namespace std;
const int N=1e6+5;
typedef long long ll;
int n,m;
int a[N],tot=0;
int v[N],tr[N][2],dist[N],siz[N];
int root[N],l[N],r[N];

inline int merge(int x,int y)
{
    if (!x||!y)return x+y;
    if (v[x]<v[y])swap(x,y);
    tr[x][1]=merge(tr[x][1],y);
    siz[x]=siz[tr[x][0]]+siz[tr[x][1]]+1;
    if (dist[tr[x][1]]>dist[tr[x][0]])swap(tr[x][0],tr[x][1]);
    dist[x]=dist[tr[x][1]]+1;
    return x;
}

inline void pop(int &x)
{
    x=merge(tr[x][0],tr[x][1]);
}
inline int newnode(int x)
{
    v[++tot]=x;
    siz[tot]=1;
    tr[tot][0]=tr[tot][1]=dist[tot]=0;
    return tot;
}
int main()
{
    scanf("%d",&n);
    fo(i,1,n)
    {
        scanf("%d",&a[i]);
        a[i]-=i;
    }
    int cnt=0;
    fo(i,1,n)
    {
        cnt++;
        root[cnt]=newnode(a[i]);
        l[cnt]=r[cnt]=i;
        while (cnt>1&&v[root[cnt]]<v[root[cnt-1]])
        {
            cnt--;
            root[cnt]=merge(root[cnt],root[cnt+1]);
            r[cnt]=r[cnt+1];
            while (siz[root[cnt]]*2>r[cnt]-l[cnt]+2)pop(root[cnt]);
        }
    }
    ll ans=0;
    fo(i,1,cnt)
    {
        int t=v[root[i]];
        fo(j,l[i],r[i])ans+=abs(t-a[j]);
    }
    printf("%lld\n",ans);
    return 0;
}
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值