枚举长度为m的所有段,尝试用中位数更新答案。

所以需要数据结构,支持查询k大,以及大于/小于 k大值 的数的和。

平衡树、权值线段树、权值分块什么的随便呢。

 #include<cstdio>
#include<algorithm>
#include<cmath>
using namespace std;
typedef long long ll;
struct Point{int v,p;}t[];
bool operator < (const Point &a,const Point &b){return a.v<b.v;}
int sumv[],ma[],en,a[],b[],tot[],l[],r[],num[],sum=,K,n,m;
ll ans=,tAns;
void makeblock()
{
int sz=sqrt(en); if(!sz) sz=;
for(;sum*sz<en;++sum)
{
l[sum]=r[sum-]+; r[sum]=sum*sz;
for(int i=l[sum];i<=r[sum];++i) num[i]=sum;
}
l[sum]=r[sum-]+; r[sum]=en;
for(int i=l[sum];i<=r[sum];++i) num[i]=sum;
}
void Insert(const int &x){++b[x]; ++tot[num[x]]; sumv[num[x]]+=(ll)ma[x];}
void Delete(const int &x){--b[x]; --tot[num[x]]; sumv[num[x]]-=(ll)ma[x];}
int Query(const int &x)
{
int cnt=; ll now=;
for(int i=;;i++)
{
cnt+=tot[i]; now+=sumv[i];
if(cnt>=x)
{
cnt-=tot[i]; now-=sumv[i];
for(int j=l[i];;j++)
{
cnt+=b[j]; now+=(ll)ma[j]*(ll)b[j];
if(cnt>=x)
{
cnt-=b[j]; now-=(ll)ma[j]*(ll)b[j];
tAns=((ll)ma[j]*(ll)cnt-now);
return j;
}
}
}
}
}
void Next_Sum(const int &x)
{
int cnt=; ll now=;
for(int i=x+;i<=r[num[x]];++i) {cnt+=b[i]; now+=(ll)ma[i]*(ll)b[i];}
for(int i=num[x]+;i<=sum;++i) {cnt+=tot[i]; now+=sumv[i];}
tAns+=(now-(ll)ma[x]*(ll)cnt);
}
int main()
{
scanf("%d%d",&n,&m); K=(m>>)+;
for(int i=;i<=n;++i)
{
scanf("%d",&t[i].v);
t[i].p=i;
} sort(t+,t+n+);
ma[a[t[].p]=++en]=t[].v;
for(int i=;i<=n;++i)
{
if(t[i].v!=t[i-].v) ++en;
ma[a[t[i].p]=en]=t[i].v;
} makeblock();
for(int i=;i<=m;++i) Insert(a[i]);
int t=Query(K); Next_Sum(t); ans=min(ans,tAns);
for(int i=m+;i<=n;++i)
{
Delete(a[i-m]); Insert(a[i]);
int t=Query(K);
Next_Sum(t);
ans=min(ans,tAns);
} printf("%lld\n",ans);
return ;
}
05-18 15:03