2024华为OD试题及答案-A044-新学校选址

题目描述

为了解新学期学生暴涨的问题,小乐村要建立所新学校，
考虑到学生上学安全问题,需要所有学生家到学校的距离最短。
假设学校和所有学生家都走在一条直线之上,请问学校建立在什么位置，
能使得到学校到各个学生家的距离和最短。

输入描述

第一行: 整数 n 取值范围 [1 ,1000 ],表示有 n户家庭。
第二行: 一组整数 m 取值范围 [0, 10000 ] ，表示每户家庭的位置，所有家庭的位置都不相同。

输出描述

一个整数，确定的学校的位置。
如果有多个位置，则输出最小的。

用例

输入	5 0 20 40 10 30
输出	20
说明	20到各个家庭的距离分别为20 0 20 10 10，总和为60，最小

输入	1 20
输出	20
说明	只有一组数据，20到20距离最小，为0

输入	2 0 20
输出	0
说明	有多个地方可选，但是0数值最小

解题思路

题目要求确定学校的位置，使得所有学生从家到学校的距离总和最短。假设学校和所有学生家都在一条直线上，这其实是一个经典的“中位数”问题。

当所有点在一维空间上时，要使得这些点到某个点的距离总和最小，这个点应该是这些点的中位数。中位数是指一个数值，数据集中有一半的数比它小，另一半数比它大。

解决步骤：

1. 输入家庭数量 nnn。
2. 输入每个家庭的位置，并存储在数组中。
3. 对数组进行排序。
4. 如果 nnn 为奇数，则中位数为排序后数组的中间值；如果 nnn 为偶数，则中位数可以取中间的两个数中的任意一个。

C++程序

#include <iostream>
#include <vector>
#include <algorithm>

int main() {
    int n;
    std::cin >> n;
    std::vector<int> positions(n);
    
    for (int i = 0; i < n; ++i) {
        std::cin >> positions[i];
    }
    
    std::sort(positions.begin(), positions.end());
    
    // 取中间位置的家庭位置作为学校位置
    int school_position = positions[n / 2];
    
    std::cout << school_position << std::endl;
    return 0;
}

Python程序

def find_school_position(n, positions):
    positions.sort()
    # 取中间位置的家庭位置作为学校位置
    school_position = positions[n // 2]
    return school_position

# 输入家庭数量
n = int(input())
# 输入每个家庭的位置
positions = list(map(int, input().split()))

# 找到学校位置
school_position = find_school_position(n, positions)
print(school_position)

解释

• 代码首先读取输入的家庭数量和每个家庭的位置。
• 将家庭位置数组进行排序。
• 取排序后的中间位置的值作为学校的位置，因为这个位置使得所有学生从家到学校的距离总和最小。