货仓选址
在一条数轴上有 N 家商店,它们的坐标分别为 A1∼AN。
现在需要在数轴上建立一家货仓,每天清晨,从货仓到每家商店都要运送一车商品。
为了提高效率,求把货仓建在何处,可以使得货仓到每家商店的距离之和最小。
输入格式
第一行输入整数 N。
第二行 N 个整数 A1∼AN。
输出格式
输出一个整数,表示距离之和的最小值。
数据范围
1≤N≤100000,
0≤Ai≤40000
输入样例:
4
6 2 9 1
输出样例:
12
时间复杂度:O(nlog2n)
算法: 贪心、绝对值不等式
思路分析:
核心思想: 绝对值不等式: |x-a| + |x-b| >= |a-b| 当x∈[a,b]时满足等式
根据货仓数量的奇偶性分类讨论
(1)货仓数量为奇数:
设货仓选址为x
S = |x-a1| + |x-a2| + |x-a3| + …… + |x-an|
S = (|x-a1| + |x-an|) + (|x-a2| + |x-an-1|) + …… + |x - a(n+1)/2| 剩余一项
S >= |a1 - an| + |a2 - an-1| + …… + |x - a(n+1)/2|
当x = a(n+1)/2时,根据绝对值不等式,x都在a和b中间,满足等于条件,且最后一项为0所以取得最小
S == |a1 - an| + |a2 - an-1| + …… + 0
(2)货仓数量为偶数:
设货仓选址为x
S = |x-a1| + |x-a2| + |x-a3| + …… + |x-an|
S = (|x-a1| + |x-an|) + (|x-a2| + |x-an-1|) + …… + (|x - a(n/2)|+|x - a(n/2+1)|)
S >= |a1 - an| + |a2 - an-1| + …… + |a(n/2+1) - a(n/2)|
当x ∈[a(n/2),a(n/2+1)]时,根据绝对值不等式,x都在a和b中间,满足等于条件
S == |a1 - an| + |a2 - an-1| + …… + |a(n/2+1) - a(n/2)|
总上所述,结合两者,x取a(n/2)路径即为最小值(绝对值不等式等于情况)
前提条件是从下标0开始存取
0 1 2 3 4
1 5 7 8 n = 4/2 = 2
1 5 6 7 8 n = 5/2 = 2
1 2 3 4
1 5 7 8 n = 2
1 5 6 7 8 n = 2 错误
代码:
#include <iostream>
#include <algorithm>
#include <cstring>
#include <cmath>
using namespace std;
typedef long long ll;
typedef pair<int, int> PII;
const int INF = 1e9;
const int N = 1e5 + 5;
int n,x,res,a[N];
int main()
{
cin>>n;
for(int i = 0; i < n; i ++)cin>>a[i];
sort(a,a+n);
x = a[n/2];
for(int i = 0; i < n; i ++)res += abs(x-a[i]);
cout<<res<<endl;
return 0;
}