牛客 货仓选址

题目描述

在一条数轴上有N家商店，它们的坐标分别为 A[1]~A[N]。现在需要在数轴上建立一家货仓，每天清晨，从货仓到每家商店都要运送一车商品。为了提高效率，求把货仓建在何处，可以使得货仓到每家商店的距离之和最小。

输入描述:

第一行一个整数N，第二行N个整数A[1]~A[N]。

输出描述:

一个整数，表示距离之和的最小值。

输入

4
6 2 9 1

输出

12

备注:

对于100%的数据:N≤100000, A[i]≤1000000

———————————————————————————————————————————

首先对所有商店进行排序，然后在数轴上选一个位置作为货仓，假设货仓左边有a家商店，右边有b家商店。若a>b，那么如果将货仓向左移动一个单位（此时a>b），总距离-a+b，总距离减小，若向右移动，那么总距离+a-b，变大；所以我们将货仓向左移动，直到a=b，因为如果a<b，那么我们就会选择把货仓向右移动，使总距离减小。综上，总距离最小的货仓位置处在左右商店个数相等的时候。所以商店个数为奇数就选在最中间的那个，偶数就选在他们中间。

#include<iostream>
#include<algorithm>
#include<cmath>
using namespace std;
const int N=1e5;
int a[N];
int main(){
    int n,i,p,ans=0;
    cin>>n;
    for(i=1;i<=n;i++){
        cin>>a[i];
    }
    sort(a,a+n);
    if(n%2==1){
        p=a[(n+1)/2];
    }
    else{
        p=a[n/2];
    }
    for(i=1;i<=n;i++){
        ans+=abs(a[i]-p);
    }
    cout<<ans;
    return 0;
}