技術標籤：資料結構資料結構演算法

貨倉選址

在一條數軸上有 N 家商店，它們的座標分別為 A1~AN。

現在需要在數軸上建立一家貨倉，每天清晨，從貨倉到每家商店都要運送一車商品。

為了提高效率，求把貨倉建在何處，可以使得貨倉到每家商店的距離之和最小。

輸入格式
第一行輸入整數N。

第二行N個整數A1~AN。

輸出格式
輸出一個整數，表示距離之和的最小值。

資料範圍

1≤N≤100000,
0≤Ai≤40000

輸入樣例：

4
6 2 9 1

輸出樣例：

12

第一種:

#include <iostream>
#include <algorithm>

using namespace std;

const int N = 100010;

int n;
int a[N];

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

第二種

#include <iostream>
#include <algorithm>

using namespace std;

const int N = 100010;

int n;
int a[N];

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

nth_element(a,a+k,a+n):函式只是把下標為k的元素放在了正確位置，對其它元素並沒有排序，當然k左邊元素都小於等於它，右邊元素都大於等於它，所以可以利用這個函式快速定位某個元素

Java

import java.util.Arrays;
import java.util.Scanner;

public class Main {
    public static void main(String[] args) {
        int n,N=100010;
        int[] a = new int[N];
        Scanner s = new Scanner
 
(System.in);
        n = s.nextInt();
        for (int i = 0; i < n; i++) {
                a[i] = s.nextInt();
        }
        Arrays.sort(a,0,n);
        int res = 0;
        for (int i = 0; i < n; i++) {
            res += Math.abs(a[i]-a[n/2]);
        }

        System.out.println(res);
    }
}