AI 刷题2 理想火车站定位问题 题解 | 豆包MarsCode AI刷题

98 阅读4分钟

理想火车站定位问题的全面分析与解决方案

问题背景

小F是A市的市长,目前面临着一项重要的城市基础设施建设任务:选择一个最优位置建设新的火车站,以便最大程度地方便市民的日常出行。A市的街道布局呈规整的网格状,这意味着每个位置之间的距离可以通过曼哈顿距离计算。曼哈顿距离是一个常用的度量方法,尤其适用于网格状城市布局,因为它通过计算两点之间在横纵坐标上的绝对差值和来反映距离。

市政府已经列出了若干个备选位置,任务是从中选择一个使得所有市民到该火车站的总旅行距离最短的位置。这是一个典型的优化问题,也是城市规划中常见的实际问题。

问题的重要性

火车站作为公共交通的重要节点,其位置的选择直接影响到市民的出行效率和城市交通的拥堵状况。一个理想的火车站位置不仅可以减少市民的通勤时间,还能促进城市经济的发展和环境的改善。因此,这个决策必须基于全面的数据分析和科学的方法。

解决方案

算法思路

解决这个问题的核心是计算每个备选火车站位置与所有市民居住地点之间的总曼哈顿距离,并找出这个总距离最小的位置。算法步骤包括:

  1. 初始化最小距离:设置一个非常大的数作为初始的最小距离,以便在后续比较中更新。
  2. 遍历所有备选位置:对于每一个备选位置,计算它到每一位市民居住位置的曼哈顿距离之和。
  3. 更新最优位置:如果当前位置的总距离小于已知的最小距离,更新最小距离并记录这个位置。
  4. 选择最优位置:遍历完成后,拥有最小总距离的位置即为最优火车站位置。

具体实现

代码实现使用C++进行,利用其强大的STL库和简洁的语法来处理数据和执行计算。程序首先定义了一个函数 solution,接受市民数量、备选位置数量、市民坐标列表和备选位置坐标列表作为输入。通过双层循环,外层循环遍历所有备选位置,内层循环计算该位置到所有市民位置的距离和。如果计算得到的总距离小于当前记录的最小距离,更新最小距离和最佳位置。

python代码

def solution(n, m, citizens, locations):
    from sys import maxsize
    min_distance = maxsize
    best_location = None

    # 遍历每个备选火车站位置
    for location in locations:
        total_distance = 0
        
        # 计算当前火车站位置到所有市民位置的曼哈顿距离之和
        for citizen in citizens:
            total_distance += abs(location[0] - citizen[0]) + abs(location[1] - citizen[1])
        
        # 如果当前总距离小于最小距离,更新最小距离和最佳位置
        if total_distance < min_distance:
            min_distance = total_distance
            best_location = location

    return best_location

# 测试用例
citizens1 = [[-1, -1], [-1, 1], [1, -1], [1, 1]]
locations1 = [[3, 2], [1, 0], [0, 0]]
print(solution(4, 3, citizens1, locations1))  # 应输出 [1, 0]

citizens2 = [[0, 0], [0, 4]]
locations2 = [[0, 2], [0, 3]]
print(solution(2, 2, citizens2, locations2))  # 应输出 [0, 2]

时间复杂度

该算法的时间复杂度为 O(M×N)O(M \times N)O(M×N),其中 MMM 是备选位置的数量,NNN 是市民的数量。这是因为算法需要对每一个备选位置计算其到所有市民位置的距离总和。

空间复杂度

空间复杂度为 O(1)O(1)O(1),除了输入数据外,算法只使用了固定的额外空间来存储最小距离和最佳位置。

测试用例和验证

提供的代码中包括了一个 main 函数用于验证算法的正确性。通过定义几组市民位置和备选位置,调用 solution 函数并输出结果,以确保算法能正确找到使总旅行距离最小的火车站位置。此外,通过一个辅助函数 vectorEqual 对比预期结果和实际输出,确保功能的准确性。

总结

选择理想火车站位置的问题是一个典型的优化问题,在城市规划和公共资源管理领域非常重要。本文提供的算法通过计算曼哈顿距离来寻找最优火车站位置,能有效地解决这一实际问题。这种方法不仅科学、高效,而且易于实现和验证。此外,理解和应用这类算法对于解决其他类似的优化问题也具有重要意义。