题目描述
为了解新学期学生暴涨的问题,小乐村要建立所新学校,
考虑到学生上学安全问题,需要所有学生家到学校的距离最短。
假设学校和所有学生家都走在一条直线之上,请问学校建立在什么位置,
能使得到学校到各个学生家的距离和最短。
输入描述
第一行: 整数 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数值最小 |
解题思路
题目要求确定学校的位置,使得所有学生从家到学校的距离总和最短。假设学校和所有学生家都在一条直线上,这其实是一个经典的“中位数”问题。
当所有点在一维空间上时,要使得这些点到某个点的距离总和最小,这个点应该是这些点的中位数。中位数是指一个数值,数据集中有一半的数比它小,另一半数比它大。
解决步骤:
- 输入家庭数量 nnn。
- 输入每个家庭的位置,并存储在数组中。
- 对数组进行排序。
- 如果 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)
解释
- 代码首先读取输入的家庭数量和每个家庭的位置。
- 将家庭位置数组进行排序。
- 取排序后的中间位置的值作为学校的位置,因为这个位置使得所有学生从家到学校的距离总和最小。