数据结构与算法学习笔记----贪心·绝对值不等式
@@ author: 明月清了个风
@@ first publish time: 2025.4.5ps⭐️感觉其实是一个数学的问题,
Acwing 104. 货仓选址
[原题链接](104. 货仓选址 - AcWing题库)
在一条数轴上有 N N N家商店,他们的坐标分别为 A 1 ∼ A N A_1 \sim A_N A1∼AN.
现在需要在数轴上建立一家货仓,每天清晨,从货仓到每家商店都要运送一车商品。
为了提高效率,求把货仓建在何处,可以使得货仓到每家商店的距离之和最小。
输入格式
第一行输入整数 N N N,
第二行 N N N个整数 A 1 ∼ A N A_1 \sim A_N A1∼AN。
输出格式
输出一个整数,表示距离之和的最小值。
数据范围
1 ≤ N ≤ 100000 1 \le N \le 100000 1≤N≤100000,
0 ≤ A i ≤ 40000 0 \le A_i \le 40000 0≤Ai≤40000
思路
将所有商店的位置记为
x
1
,
x
2
,
⋯
,
x
n
x_1, x_2, \cdots,x_n
x1,x2,⋯,xn,货仓的位置为$x, 那么距离之和就是
f
(
x
)
=
∣
x
1
−
x
∣
+
∣
x
2
−
x
∣
+
⋯
+
∣
x
n
−
1
−
x
∣
+
∣
x
n
−
x
∣
f(x) = |x_1 - x| + |x_2 - x| +\cdots + |x_{n - 1} - x| + |x_n - x|
f(x)=∣x1−x∣+∣x2−x∣+⋯+∣xn−1−x∣+∣xn−x∣
那么对于题目就转化为求
f
(
x
)
f(x)
f(x)的最小值,将式(1)移项变为两两一对的表示方法
f
(
x
)
=
(
∣
x
1
−
x
∣
+
∣
x
n
−
x
∣
)
+
(
∣
x
2
−
x
∣
+
∣
x
n
−
1
−
x
∣
)
+
⋯
f(x) = (|x_1 - x| + |x_n - x|) + (|x_2 - x| +|x_{n - 1} - x|) + \cdots
f(x)=(∣x1−x∣+∣xn−x∣)+(∣x2−x∣+∣xn−1−x∣)+⋯
对于每一个括号里的值来说的最小值就是
∣
x
n
−
x
1
∣
|x_n - x_1|
∣xn−x1∣,当
x
x
x满足
x
1
≤
x
≤
x
n
x_1 \le x \le x_n
x1≤x≤xn即可,对于任意一组都满足
x
x
x在两个端点之间,就可以推出
x
x
x一定在所有点的中间位置。
放在代码的处理上就可以简单一点,将这个位置放在中间的一组的两个商店的某一个店上就行,可以直接取 a [ n / 2 ] a[n / 2] a[n/2]的位置。
代码
#include <iostream>
#include <cstring>
#include <cstdio>
#include <algorithm>
using namespace std;
const int N = 100010;
int n;
int a[N];
int main()
{
cin >> n;
for(int i = 0; i < n; i ++) cin >> a[i];
sort(a, a + n);
int res = 0;
for(int i = 0; i < n; i ++) res += abs(a[i] - a[n / 2]);
cout << res << endl;
return 0;
}