一、题目
在一条数轴上有 N 家商店,它们的坐标分别为 A1∼AN。
现在需要在数轴上建立一家货仓,每天清晨,从货仓到每家商店都要运送一车商品。
为了提高效率,求把货仓建在何处,可以使得货仓到每家商店的距离之和最小。
输入格式
第一行输入整数 N。
第二行 N 个整数 A1∼AN。
输出格式
输出一个整数,表示距离之和的最小值。
数据范围
1≤N≤100000。
0≤Ai≤40000。
输入样例:
4
6 2 9 1
输出样例:
12
难度:简单 |
时/空限制:1s / 64MB |
总通过数:42865 |
总尝试数:66652 |
来源:《算法竞赛进阶指南》, 模板题 |
算法标签 |
二、思路
1.公式的推导
基本:在标有a,b两个数的这条轴选择一个x,使得x到a的距离加上x到b的距离的和最小
|x-a|+|x-b|>=|a-b|
ps:>=的原因是,若x在a,b之间则=,若x在a,b之外则>;
同理推导到n个点上:
a[1],a[2]...a[n](假设数组a已经从小到大排好序)
则可以两两配对,得出另一个推导公式
|x-a[1]|+|x-a[2]|+...+|x-a[n]|>=|a[n]-a[1]|+|a[n-1]-a[2]|+...
若想要其和最小,则应该取等号
在两两配对的情况下:
1.若n为奇数,则x取中位数
2.若n为偶数则x取最后配对的两个数的中间的值
2.方法1:排序+绝对值不等式 时间复杂度 o(nlogn)
cin>>n;
int i,j;
for(i=0;i<n;i++)
cin>>a[i];
sort(a,a+n);
int ans=0;
for(i=0;i<n;i++)
ans+=abs(a[i]-a[n/2]);//ans+=a[i]-a[i/2]
cout<<ans<<endl;
3.方法二:利用nth_element()函数(不用排序)时间复杂度:o(n)
cin>>n;
int i;
for(i=0;i<n;i++)
cin>>a[i];
nth_element(a,a+n/2,a+n);
int ans=0;
for(i=0;i<n;i++)
ans+=abs(a[i]-a[n/2]);
cout<<ans<<endl;
nth_element()用法
nth_element(数组名,数组名+第k小的元素,数组名+元素个数)
时间复杂度o(n)
作用是将第k小的数放在区间第k个位置
4.此题的拓展
1.一维扩展二维(三分) 3167星星还是树(acwing)
2.多维 模拟退火算法
三、完整代码
方法一:
#include<bits/stdc++.h>
#define endl '\n'
#define int long long
#define IOS ios::sync_with_stdio(false); cin.tie(0); cout.tie(0);
using namespace std;
const int N=2e5+10;
const int inf=0x3f3f3f3f;
int n;
int a[N];
void solve()//n(logn)
{
cin>>n;
int i,j;
for(i=0;i<n;i++)
cin>>a[i];
sort(a,a+n);
int ans=0;
for(i=0;i<n;i++)
ans+=abs(a[i]-a[n/2]);//ans+=a[i]-a[i/2]
cout<<ans<<endl;
}
signed main()
{
int t=1;
while(t--)
{
solve();
}
return 0;
}
方法二:
#include<bits/stdc++.h>
#define endl '\n'
#define int long long
#define IOS ios::sync_with_stdio(false); cin.tie(0); cout.tie(0);
using namespace std;
const int N=2e5+10;
const int inf=0x3f3f3f3f;
int n;
int a[N];
void solve()//o(n)
{
cin>>n;
int i;
for(i=0;i<n;i++)
cin>>a[i];
nth_element(a,a+n/2,a+n);
int ans=0;
for(i=0;i<n;i++)
ans+=abs(a[i]-a[n/2]);
cout<<ans<<endl;
}
signed main()
{
int t=1;
while(t--)
{
solve();
}
return 0;
}