贪心问题-货仓选址

题目

https://www.acwing.com/activity/content/problem/content/1214/

题目分析

假设出的解决方案：将所有商店在坐标上排序后取中位数位置放置货舱。如果是奇数个商店则货舱位置为排序后中间位置的商店，如果是偶数个商店则货舱位置为中间两个商店之间的任意位置。

方案证明

奇数个商店

共 n 个商店选中 n/2(向下取整) + 1 个商店为货舱位置后，左边送货的距离总和为 p，右边送货的距离总和为 q，距离总和为 p + q

将货舱左移一位，左边送货的距离总和变为 p - n/2 ，右边送货的距离总和变为 q + n/2 + 1，距离总和变为 p + q + 1，距离总和变大

同理往右边挪一位相同，都会多出一个距离中间商店的送货距离。

偶数个商店

如果货舱选择的位置在 n / 2 与 n / 2 + 1 个商店之间不管怎么移动距离总和会保持不变，只会改变货舱到各个商店的相对位置，将货舱位置选定到 n / 2 的位置上，则货舱左边的送货距离为 p，右边的送货距离为 q

将货舱左移一位，左边的送货距离变为为 p - ( n/2 - 1)，右边的送货距离变为 p + n/2 + 1，距离总和变为 p + q + 2，距离综合变大

同理当货舱位置选定到 n / 2 + 1 位置时，向右移一位都会多出到中间两个货舱的距离。

发现无论商店数量的奇偶，都可以选择 n / 2 的位置为货舱位置。

证毕！

代码实现

数据范围

最多有1e5个商店，商店的坐标最大是4e4。去极限，前50001个商店的坐标为0，其余所有商店坐标为4e4，最大的距离和为49999 * 4e4 < 50000 * 4e4 = 2e8，int 的数据范围为 2.1e9，没有爆 int 可以使用 int 存储距离。

实现思路

使用数组存储所有商店的坐标，排序后存储每个商店到 n / 2 位置商店的距离即可。

代码实现

#include<iostream>
#include<algorithm>

using namespace std ;

const int N = 1e6 + 10 ;

int n ;
int a[N] ;

int main()
{
    cin >> n ;
    for(int i = 0 ; i < n ; i ++)
        scanf("%d", &a[i]) ;
    
    sort(a, a + n) ;
    
    int sum = 0 ;
    for(int i = 0 ; i < n ; i ++)
        sum += abs(a[i] - a[n / 2]) ;
    
    printf("%d", sum) ;
    
    return 0 ;
}