题目来源:码蹄集
题目描述:
算法思路:
1. 对居民的位置进行排序;
2. 取中位数作为外卖站的位置,因为到每个居民的距离之和最小,必然在某个居民的位置或者两个居民的位置之间;
3. 计算每个居民到外卖站的距离之和。
B站老师思路讲解链接:https://www.bilibili.com/video/BV1MX4y1R77W/?t=519.4&vd_source=3ae2a916df1bc5c1114c2bf3e95a2118
C++代码实现:
#include <iostream>
#include <algorithm>
using namespace std;
int main() {
int n;
cin >> n;
int a[n];
for (int i = 0; i < n; i++) {
cin >> a[i];
}
sort(a, a + n);
// 如果 n 为奇数,则站点建在最中间的居民处最优
if (n % 2 == 1) {
int center = a[n / 2];
long long dist = 0;
// 统计到每个居民的距离之和
for (int i = 0; i < n; i++) {
dist += abs(a[i] - center);
}
cout << dist << endl;
}
// 如果 n 为偶数,则站点建在最中间的两个居民中任意一个处最优
else {
int left_center = a[n / 2 - 1];
int right_center = a[n / 2];
long long dist_left = 0, dist_right = 0;
// 计算左右两个站点的距离之和,并取其较小值
for (int i = 0; i < n; i++) {
dist_left += abs(a[i] - left_center);
dist_right += abs(a[i] - right_center);
}
cout << min(dist_left, dist_right) << endl;
}
return 0;
}
Java代码实现:
import java.util.Arrays;
import java.util.Scanner;
public class Main {
public static void main(String[] args) {
Scanner scanner = new Scanner(System.in);
int n = scanner.nextInt();
int[] a = new int[n];
for (int i = 0; i < n; i++) {
a[i] = scanner.nextInt();
}
Arrays.sort(a);
// 如果 n 为奇数,则站点建在最中间的居民处最优
if (n % 2 == 1) {
int center = a[n / 2];
long dist = 0;
// 统计到每个居民的距离之和
for (int i = 0; i < n; i++) {
dist += Math.abs(a[i] - center);
}
System.out.println(dist);
}
// 如果 n 为偶数,则站点建在最中间的两个居民中任意一个处最优
else {
int left_center = a[n / 2 - 1];
int right_center = a[n / 2];
long dist_left = 0, dist_right = 0;
// 计算左右两个站点的距离之和,并取其较小值
for (int i = 0; i < n; i++) {
dist_left += Math.abs(a[i] - left_center);
dist_right += Math.abs(a[i] - right_center);
}
System.out.println(Math.min(dist_left, dist_right));
}
}
}
Python代码实现(会错一个测试用例):
n = int(input())
a = list(map(int, input().split()))
a.sort()
mid = (a[n // 2] + a[(n - 1) // 2]) // 2 # 取中位数
ans = sum(abs(x - mid) for x in a)
print(ans)