题意
在数轴上有 \(n\) 个点,有 \(n\) 个人和一辆车。每个人要由车送去一个点。
车一次最多能载 \(k\) 个人。送完人后不用返回。求最小距离。
Sol
将数轴左边与右边的点分开处理。
每次都处理当前最靠右(左)且没到过的点。
那么开两个数组分别存下正坐标和负坐标,暴力枚举即可。
最后记得把返程的路减掉。
时间复杂度 \(O(n \log n)\),瓶颈在于排序。
Code
//LYC_music yyds!
#include<bits/stdc++.h>
#define IOS ios::sync_with_stdio(0)
#define lowbit(x) (x&(-x))
#define int long long
using namespace std;
inline char gc()
{
static char buf[1000000],*p1=buf,*p2=buf;
return p1==p2&&(p2=(p1=buf)+fread(buf,1,1000000,stdin),p1==p2)?EOF:*p1++;
}
int read()
{
int pos=1,num=0;
char ch=getchar();
while (!isdigit(ch))
{
if (ch=='-') pos=-1;
ch=getchar();
}
while (isdigit(ch))
{
num=num*10+(int)(ch-'0');
ch=getchar();
}
return pos*num;
}
void write(int x)
{
if (x<0)
{
putchar('-');
write(-x);
return;
}
if (x>=10) write(x/10);
putchar(x%10+'0');
}
void writesp(int x)
{
write(x);
putchar(' ');
}
void writeln(int x)
{
write(x);
putchar('\n');
}
const int N=2e5+10;
int n,m,n1,n2,ans,a[N],b[N];
signed main()
{
int T=read();
while (T--)
{
n=read(); m=read(); n1=n2=ans=0;
for (int i=1;i<=n;i++)
{
int x=read();
if (x>0) a[++n1]=x;
else b[++n2]=-x;
}
sort(a+1,a+n1+1);
sort(b+1,b+n2+1);
for (int i=n1;i>=1;i-=m)
ans+=a[i]*2;
for (int i=n2;i>=1;i-=m)
ans+=b[i]*2;
if (n1&&n2) ans-=max(a[n1],b[n2]);
else if (n1) ans-=a[n1];
else if (n2) ans-=b[n2];
writeln(ans);
}
}