题目链接:货仓选址
题目大意:
在一条数轴上有 N 家商店,它们的坐标分别为 A1∼AN。
现在需要在数轴上建立一家货仓,每天清晨,从货仓到每家商店都要运送一车商品。
为了提高效率,求把货仓建在何处,可以使得货仓到每家商店的距离之和最小。
解题思路:
第一种(前缀和):
根据题目数据可知货仓的位置范围为0 ~ 40000,所以我们想到可以用O(n)的复杂度从0 ~ 40000枚举每个位置,枚举每个位置的时候求出该位置到所有商店的距离之和,然后求一个最小值。关键在于如何用O(1)的复杂度更新货仓到所有商店的距离之和。在此之前我们先用前缀和求出i位置之前一共有几家商店。我们用一个变量sum记录所有商店到货仓的距离之和,然后假设从i位置移动到i+1位置,那么小于等于i位置的所有商店到货仓的距离就+1,大于i位置的所有商店到货仓的距离就-1,此时sum = sum + s[ i ] - (n - s[ i ]),这样我们就用O(1)的时间更新了sum,对所有的sum取一个最小值即可。
拓展: 假如商店的范围为0 ~ 1e9,那么我们该怎么做哪,对所有的Ai进行一个 离散化 即可。
第二种(三分):
根据第一种思路可知我们在枚举每个位置时,sum是先变小再变大的,所有我们不妨设自变量x为货仓的位置,因变量y是所有商店到货仓的距离之和,那么函数图像就是一个类似于开口向上的二次函数,进而可以看出符合三分的性质。
第三种(中位数):
我们将n个商店的位置从小到大排序,并从1 ~ n进行编号。
假设n=1,那么货仓建在1号商店的位置最优,答案为0。
假设n=2,那么货仓建在1号商店和2号商店之间的位置最优,答案为 x2 - x1。
假设n=3,那么货仓建在2号商店的位置最优,答案为 x3 - x1。
假设n=4,那么货仓建在2号商店和3号商店之间的位置最优,答案为 x4 - x1 + x3 - x2。
假设n=5,那么货仓建在3号商店的位置最优,答案为 x5 - x1 + x4 - x2。
… …
综上,货仓建在(1 + n) / 2 (中位数) 号商店的位置最优。
AC代码:
第一种:
#include <bits/stdc++.h>
using namespace std;
typedef long long ll;
typedef pair<int, int> PII;
const int N = 1e5 + 10;
int a[N], s[N];
int main()
{
int n; cin >> n;
ll sum = 0;
for (int i = 1; i <= n; i++)
{
int x; cin >> x;
sum += x;
a[x]++;
}
s[0] = a[0];
for (int i = 1; i <= 4e4; i++) s[i] = s[i - 1] + a[i];
ll ans = sum;
for (int i = 0; i <= 4e4; i++)
{
int l = s[i], r = n - s[i];
sum = sum + l - r;
ans = min(ans, sum);
}
cout << ans << endl;
return 0;
}
第二种:
#include <bits/stdc++.h>
using namespace std;
typedef long long ll;
typedef pair<int, int> PII;
const int N = 1e5 + 10;
int n, a[N];
ll check(int x)
{
ll ans = 0;
for (int i = 1; i <= n; i++) ans += abs(x - a[i]);
return ans;
}
int main()
{
cin >> n;
for (int i = 1; i <= n; i++) cin >> a[i];
int l = 0, r = 4e4;
while (r - l > 1)
{
int mid = l + r >> 1;
int midl = mid - 1, midr = mid + 1;
if (check(midl) < check(midr)) r = mid;
else l = mid;
}
cout << min(check(l), check(r)) << endl;
return 0;
}
第三种:
#include <bits/stdc++.h>
using namespace std;
typedef long long ll;
typedef pair<int, int> PII;
const int N = 1e5 + 10;
int a[N], n;
int main()
{
cin >> n;
for (int i = 1; i <= n; i++) cin >> a[i];
nth_element(a + 1, a + (1 + n) / 2, a + 1 + n);
ll ans = 0;
for (int i = 1; i <= n; i++) ans += abs(a[i] - a[(1 + n) / 2]);
cout << ans << endl;
return 0;
}