【校招刷题】HZOJ-45仓库选址

仓库选址

题目描述

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

货仓选址 - 题目 - Online Judge (haizeix.com)

输入

第一行输入一个数 N。（1≤N≤100000）
接下来一行，输入 N 个数，表示商店的坐标。

输出

输出最短距离之和。

样例输入

5
1 3 5 6 10
样例输出
12

数据规模与约定

时间限制：1 s

内存限制：256 M

100% 的数据保证 1≤N≤100000

#include <iostream>
#include <vector>
#include <bits/stdc++.h>

using namespace std;
int main() {
    int n;
    vector<int> arr;
    cin>>n;
    for(int i=0,a;i<n;i++){
        cin>>a;
        arr.push_back(a);
    }
    sort(arr.begin(),arr.end());
    int p=arr[n/2];
    int sum=0;
    for(int i=0;i<n;i++){
        sum+=abs(arr[i]-p);
    }
    cout<<sum;
    return 0;
}

问题分析：

为什么说选取中位数的商店是总距离最小的选址呢？

我们假设选取一点P作为选址，假设该P点左边有A个商店，右边有B个商店，且A>B，那么要使总距离sum减少，我们要做的是将P点往左移一点点距离d，那么左边减少的距离为Ad，左边增加的距离是Bd，那么sum’=sum+(B-A)*d,sum’就比sum小，那么能不能一直左移呢？显然不行，因为当移到B>A时，sum就会增大了，sum最小时位置是中位数的点。