似乎和前几天做那道IOI的题有点相似?都是维护中位数。
不过IOI的题良心多了QAQ,坐标是递增的所以直接线性维护就好了。。然而。。这个显然需要维护一个区间K大,且支持插入删除操作。
好久没打splay手生。。删除后忘记上传标记T了好久。。
果然常数巨大,被各路treap和set(卧槽)虐成狗了。
#include<bits/stdc++.h>
#define inf 1e18
#define ll long long
#define N 100005
using namespace std;
int n,k,t1,t2,cnt,root;
ll ans=inf;
int size[N],h[N],fa[N],num[N],tree[N][2];
ll sum[N];
inline int read()
{
int a=0,f=1; char c=getchar();
while (c<'0'||c>'9') {if (c=='-') f=-1; c=getchar();}
while (c>='0'&&c<='9') {a=a*10+c-'0'; c=getchar();}
return a*f;
}
inline void pushup(int k)
{
size[k]=size[tree[k][0]]+size[tree[k][1]]+1;
sum[k]=sum[tree[k][0]]+sum[tree[k][1]]+(ll)num[k];
}
inline void rotate(int x,int &k)
{
int y=fa[x],z=fa[y],l=tree[y][1]==x,r=l^1;
if (y==k) k=x; else tree[z][tree[z][1]==y]=x;
fa[x]=z; fa[y]=x; fa[tree[x][r]]=y;
tree[y][l]=tree[x][r]; tree[x][r]=y;
pushup(y); pushup(x);
}
inline void splay(int x,int &k)
{
while (x!=k)
{
int y=fa[x],z=fa[y];
if (y!=k)
{
if (tree[y][0]==x^tree[z][0]==y) rotate(x,k); else rotate(y,k);
}
rotate(x,k);
}
}
void insert(int &k,int x,int f)
{
if (!k)
{
k=++cnt;
fa[k]=f;
num[k]=x;
size[k]=1;
splay(k,root);
return;
}
if (x<num[k]) insert(tree[k][0],x,k); else insert(tree[k][1],x,k);
}
int ask_kth(int k,int rank)
{
if (!k) return -1;
if (size[tree[k][0]]+1==rank) return k;
else if (size[tree[k][0]]+1>rank) return ask_kth(tree[k][0],rank);
else return ask_kth(tree[k][1],rank-size[tree[k][0]]-1);
}
int main()
{
n=read(); k=read();
for (int i=1;i<=n;i++) h[i]=read();
for (int i=1;i<k;i++)
insert(root,h[i],0);
for (int i=1;i<=n-k+1;i++)
{
int t;
insert(root,h[i+k-1],0);
t=ask_kth(root,(k+1)>>1);
splay(t,root);
ll tot=(ll)num[t]*size[tree[t][0]]-sum[tree[t][0]];
tot+=sum[tree[t][1]]-(ll)num[t]*size[tree[t][1]];
ans=min(ans,tot);
splay(i,root);
if (tree[i][0])
{
root=tree[i][0];
int x=tree[i][0];
while (tree[x][1]) x=tree[x][1];
fa[tree[i][1]]=x; tree[x][1]=tree[i][1]; fa[tree[i][0]]=0;
tree[i][0]=tree[i][1]=0;
while (x) pushup(x),x=fa[x];
}
else fa[tree[i][1]]=0,root=tree[i][1],tree[i][1]=0;
}
cout << ans;
}