AcWing 104. 货仓选址（绝对值不等式、排序、贪心）

一、题目

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