zl程序教程

您现在的位置是:首页 >  其它

当前栏目

外卖递送 码蹄集

外卖
2023-09-11 14:16:29 时间

题目来源:码蹄集

题目描述:

在这里插入图片描述
在这里插入图片描述

算法思路:

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)

代码提交测试结果:

在这里插入图片描述