题目链接: http://codeforces.com/group/NVaJtLaLjS/contest/241382/problem/F
题意: 在x轴上,分布了n个物品,问你在x轴某个地方建一个仓库收集起来,可以使得收集完货车走的总路程最短,一开始货车从仓库出发,每次最多能运输m个物品
思路: 找到这批物品的中位数即可,中位数所在地方就是仓库最近的。
但是后来我在写代码时犯了一些小错误,
一开始我是这样写的
for(int i=pos-1;i>0;i-=m)
{
sum+=a[i]-a[pos];
}
for(int i=pos1;i<=n;i+=m)
{
sum+=a[pos]-a[i];
}
但是这样的话就会导致最后未满足m个物品被跳过结束循环,而被没有被运到仓库
所以我们可以从远处开始
for(int i=n;i>pos;i-=m)
{
sum+=a[i]-a[pos];
}
for(int i=1;i<pos;i+=m)
{
sum+=a[pos]-a[i];
}
#include<bits/stdc++.h>
using namespace std;
const int N=1e6+5;
typedef long long ll;
ll a[N];
int main()
{
ll n,m;
ll sum=0;
scanf("%lld%lld",&n,&m);
for(int i=1;i<=n;i++)
{
scanf("%lld",&a[i]);
}
ll pos=(n+1)/2;
for(int i=n;i>pos;i-=m)
{
sum+=a[i]-a[pos];
}
for(int i=1;i<pos;i+=m)
{
sum+=a[pos]-a[i];
}
cout<<sum*2<<endl;
return 0;
}