货仓选址(前缀和 or 三分 or 中位数)

题目链接:货仓选址

题目大意:

在一条数轴上有 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;
}

  • 0
    点赞
  • 2
    收藏
    觉得还不错? 一键收藏
  • 0
    评论

“相关推荐”对你有帮助么?

  • 非常没帮助
  • 没帮助
  • 一般
  • 有帮助
  • 非常有帮助
提交
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值