算法-小美走公路

题目

有一个环形的公路，上面共有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

输出描述：
一个正整数，代表小美走的最短距离。
 

思路

建立一个2n+1大小的数组，存放把每一个距离都存放两次，分别为1-n和n+1-2n。这样就可以在一个数组里完成顺时针和逆时针的变换。

注意：x和y的大小是不定的，要想像下面的代码一样用y-x，要先使x成为比较小的站点。

代码

#include <bits/stdc++.h>
using namespace std;

int main() {
    int n;
    long long x, y, a;
    cin >> n;

    vector<long long> dis(2*n + 1);
    for(int i = 1; i <= n; i++){
        cin >> a;
        dis[i] = a;
        dis[i + n] = a;
    }

    cin >> x >> y;
    if(x > y) swap(x, y);

    // 顺时针
    long long dis1 = 0, dis2 = 0;
    for(int i = x; i < y; i++){
        dis1 += dis[i];
    }

    // 逆时针
    for(int i = y; i < x + n; i++){
        dis2 += dis[i];
    }
    
    cout << min(dis1, dis2);

    return 0;
}