大厂真题：【模拟】美团2023秋招-小美走公路

题目描述与示例

题目描述

有一个环形的公路，上面共有n站，现在给定了顺时针第i站到第i`` ``+`` ``1站之间的距离（特殊的，也给出了第n站到第1站的距离）。小美想沿着公路第x站走到第y站，她想知道最短的距离是多少？

输入描述

第一行输入一个正整数n，代表站的数量。

第二行输入n个正整数ai，前n`` ``-`` ``1个数代表顺时针沿着公路走，i站到第i`` ``+`` ``1站之间的距离；最后一个正整数代表顺时针沿着公路走，第n站到第1站的距离。

第三行输入两个正整数x和y，代表小美的出发地和目的地。

1 <= n <= 10^5
1 <= a <= 10^9
1 <= x, y <= n

输出描述

一个正整数，代表小美走的最短距离。

示例一

输入

3
1 2 2
2 3

输出

2

说明

暂时无法在飞书文档外展示此内容

示例二

输入

3
1 2 2
1 3

输出

2

说明

暂时无法在飞书文档外展示此内容

时空限制

时间限制： 3000MS

内存****限制： 589824KB

解题思路

注意，本题和LeetCode1184.公交站间的距离完全一致。

题目要求计算最短距离，实际上一共只有两条可能的路径：顺时针从x走到y和逆时针从x走到y。

因此只需要计算顺时针和逆时针的两个结果进行比较即可。

唯一需要注意的地方就是边界，即第n站到第1站的距离的计算，这里可以通过逻辑判断是否到达边界，也可以通过取余的方式避免边界问题。

代码

Python

# 题目：【模拟】美团2023秋招-小美走公路
# 作者：闭着眼睛学数理化
# 算法：模拟/数组
# 代码有看不懂的地方请直接在群上提问


n = int(input())
distance = list(map(int, input().split()))
x, y = map(int, input().split())


# 令x是索引在前的车站，y是索引在后的车站
# 另外，由于题目所给的车站索引是从1开始的
# 为了能够映射到数组distance中的索引
# 需要进行-1的操作
x, y = (x-1, y-1) if x < y else (y-1, x-1)
# 求两段距离
# 第一段为：从x顺时针走到y，对应的距离为sum(distance[x:y])
# 第二段为：从y顺时针走到x，对应的距离为sum(distance[:x]) + sum(distance[y:])
# 两者取较小值，即为答案
print(min(sum(distance[x:y]), sum(distance[:x]) + sum(distance[y:])))

Java

// 题目：【模拟】美团2023秋招-小美走公路
// 作者：助教老师-大赛
// 算法：模拟/数组
// 代码有看不懂的地方请直接在群上提问


import java.util.Scanner;

public class Main {
    public static void main(String[] args) {
        Scanner scanner = new Scanner(System.in);
        int n = scanner.nextInt(); // 读取站的数量
        int[] distances = new int[n]; // 存储各站之间的距离
        for (int i = 0; i < n; i++) {
            distances[i] = scanner.nextInt(); // 读取各站之间的距离
        }
        int x = scanner.nextInt(); // 读取出发站
        int y = scanner.nextInt(); // 读取目的站

        int clockwiseDistance = 0; // 顺时针方向的距离
        int counterClockwiseDistance = 0; // 逆时针方向的距离

        // 计算顺时针距离
        for (int i = x - 1; i != y - 1; i = (i + 1) % n) {
            clockwiseDistance += distances[i];
        }

        // 计算逆时针距离
        for (int i = y - 1; i != x - 1; i = (i + 1) % n) {
            counterClockwiseDistance += distances[i];
        }

        // 输出最短距离
        System.out.println(Math.min(clockwiseDistance, counterClockwiseDistance));
    }
}

C++

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

using namespace std;

int main() {
    int n;
    cin >> n;

    vector<int> distance(n);

    for (int i = 0; i < n; i++) {
        cin >> distance[i];
    }

    int x, y;
    cin >> x >> y;

    if (x > y) {
        swap(x, y);
    }

    long long sum1 = 0;
    long long sum2 = 0;

    for (int i = x - 1; i < y - 1; i++) {
        sum1 += distance[i];
    }

    for (int i = 0; i < x - 1; i++) {
        sum2 += distance[i];
    }

    for (int i = y - 1; i < n; i++) {
        sum2 += distance[i];
    }

    cout << min(sum1, sum2) << endl;

    return 0;
}

时空复杂度

时间复杂度：O(``n``)。求和操作需要一次遍历数组。

空间复杂度：O(``1``)。仅需若干常数变量。

华为OD算法/大厂面试高频题算法练习冲刺训练

• 华为OD算法/大厂面试高频题算法冲刺训练目前开始常态化报名！目前已服务100+同学成功上岸！

• 课程讲师为全网50w+粉丝编程博主@吴师兄学算法 以及小红书头部编程博主@闭着眼睛学数理化

• 每期人数维持在20人内，保证能够最大限度地满足到每一个同学的需求，达到和1v1同样的学习效果！

• 60+天陪伴式学习，40+直播课时，300+动画图解视频，300+LeetCode经典题，200+华为OD真题/大厂真题，还有简历修改、模拟面试、专属HR对接将为你解锁

• 可上全网独家的欧弟OJ系统练习华子OD、大厂真题

• 可查看链接 OD算法冲刺训练课程表 & OD真题汇总(持续更新)

• 绿色聊天软件戳 od1336了解更多