题目传送门
做完这题推荐3190。升级版
解法:
单调栈维护下凸包
先按直线的k来排序
当前直线为i,栈顶为x,栈顶下一个为y。
如果i和x的交点在x和y的交点的左边的话。
那么x是看不到的。踢掉。
最后剩下的就是答案。
代码实现:
#include<cstdio>
#include<iostream>
#include<algorithm>
using namespace std;
typedef long long ll;
struct node {int c,lc,rc;ll s;}t[2000010];int cnt,rt[110000];
void build(int &u,int l,int r,int p,int s) {
if(u==0)u=++cnt;t[u].c++;t[u].s+=(ll)s;
if(l==r)return ;int mid=(l+r)/2;
if(p<=mid)build(t[u].lc,l,mid,p,s);
else build(t[u].rc,mid+1,r,p,s);
}
void Merge(int &u1,int u2) {
if(u1==0) {u1=u2;return ;}if(u2==0)return ;
t[u1].c+=t[u2].c;t[u1].s+=t[u2].s;
Merge(t[u1].lc,t[u2].lc);Merge(t[u1].rc,t[u2].rc);
}
int find_mid(int &u1,int &u2,int l,int r,int k) {
if(l==r)return l;int mid=(l+r)/2;
int c=t[t[u1].lc].c-t[t[u2].lc].c;
if(k<=c)return find_mid(t[u1].lc,t[u2].lc,l,mid,k);
else return find_mid(t[u1].rc,t[u2].rc,mid+1,r,k-c);
}
int find_lc(int &u1,int &u2,int l,int r,int p) {
if(t[u1].c-t[u2].c==0)return 0;
if(l==r) return t[u1].c-t[u2].c;int mid=(l+r)/2;
if(p<=mid)return find_lc(t[u1].lc,t[u2].lc,l,mid,p);
else return find_lc(t[u1].rc,t[u2].rc,mid+1,r,p)+t[t[u1].lc].c-t[t[u2].lc].c;
}
ll find_ls(int &u1,int &u2,int l,int r,int p) {
if(t[u1].s-t[u2].s==0) return 0;
if(l==r)return t[u1].s-t[u2].s;int mid=(l+r)/2;
if(p<=mid)return find_ls(t[u1].lc,t[u2].lc,l,mid,p);
else return find_ls(t[u1].rc,t[u2].rc,mid+1,r,p)+t[t[u1].lc].s-t[t[u2].lc].s;
}
struct edge {int x,id;}a[110000];ll s[110000];
bool cmp(edge n1,edge n2){return n1.x<n2.x;}
bool cmp1(edge n1,edge n2){return n1.id<n2.id;}
int main() {
//freopen("akc.in","r",stdin);freopen("akc.out","w",stdout);
int n,k;scanf("%d%d",&n,&k);int mmax=0;
for(int i=1;i<=n;i++) {scanf("%d",&a[i].x);a[i].id=i;}
if(k==1){printf("0\n");return 0;}
sort(a+1,a+1+n,cmp);int tot=0;a[0].x=-1;int tt=0;
for(int i=1;i<=n;i++) if(a[i].x!=a[i-1].x)tt++;
for(int i=1;i<=n;i++) {if(a[i].x!=a[i-1].x)tot++;build(rt[a[i].id],1,tt,tot,a[i].x);s[tot]=a[i].x;}
for(int i=1;i<=n;i++) Merge(rt[i],rt[i-1]);
sort(a+1,a+1+n,cmp1);ll T=0;for(int i=1;i<k;i++)T+=ll(a[i].x);ll Ans=ll(100000000000);a[0].x=0;
for(int i=k;i<=n;i++) {
T+=ll(a[i].x);T-=ll(a[i-k].x);
int p=find_mid(rt[i],rt[i-k],1,tt,(k+1)/2);
int lc=find_lc(rt[i],rt[i-k],1,tt,p),rc=k-lc;
ll ls=find_ls(rt[i],rt[i-k],1,tt,p),rs=T-ls;
Ans=min(Ans,ll(lc*s[p])-ls+rs-ll(rc*s[p]));
}printf("%lld\n",Ans);
return 0;
}