距离问题、sort排序用法

题目描述

在一条数轴上有N家商店，它们的坐标分别为 A[1]~A[N]。现在需要在数轴上建立一家货仓，每天清晨，从货仓到每家商店都要运送一车商品。为了提高效率，求把货仓建在何处，可以使得货仓到每家商店的距离之和最小。

输入描述:

第一行一个整数N，第二行N个整数A[1]~A[N]。

输出描述:

一个整数，表示距离之和的最小值。

示例1

输入：

4

6 2 9 1

输出：

12

备注：

对于100%的数据：N <= 100000, A[i]  <= 1000000

解1：

思路：

假设把货仓建在第k个商店的坐标上，那么左边有 k - 1 个商店，右边有 n - k - 1 个商店。

当k<n/2时，向右移一位，因为k-1<n-k-1，所以+(k-1)-(n-k-1)会使总距离减小，因此我们应当把k往右移，直到当 k-1 = n-k-1时，若再往右移，k-1>n-k-1，+(k-1)-(n-k-1)会使总距离变大，所以不能往右移了。 若k-1 = n-k-1, 那么k=n/2, k也就是整个序列中位数的位置。

如果商店数为偶数的话，中点有两个商店，此时在这两个商店之间任选一点都可，即a[n/2]<=k<=a[n/2+1]，因为除这两个商店外左右两边的商店数是相等的（即k向右移总距离加减数相等），而这两个商店到k的距离和又是定值，所以不影响答案，我们可以直接选择a[n/2]

AC代码：

#include<bits/stdc++.h>

using namespace std;

const int N = 1e5 + 10;
int a[N]; 

int main()
{
    int i;
    int n,ans,mid;
    ans = 0;
    cin >> n;
    for(i = 1; i <= n ; i ++)  cin >> a[i];
    sort(a + 1,a + n + 1);
    mid = a[n / 2];
    for(i = 1 ; i <= n ; i ++ )  ans += abs(a[i] - mid);
    cout << ans << endl;
    return 0;
}

解2：

#include<bits/stdc++.h>
 
using namespace std;
 
const int N = 1e7 + 10;
int A[N];
int a[N];

int main()
{
    long long i,n;
    long long sum = 0;
    cin >> n;
    for(i = 0;i < n;i ++)
    {
        cin >> A[i];
    }
    sort(A , A + n );
    for(i = n - 1;i >= 0;i --)
    {
        a[n - i - 1] = A[i];
    }
    for(i = 0;A[i] < a[i];i ++)
    {
        sum += a[i] - A[i];
    }
    cout << sum << endl;
    return 0;
}