Spring VI F - Police Patrol

题目链接: 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;
}

猜你喜欢

转载自blog.csdn.net/qq_42819598/article/details/88873009
vi