题目
The task is really simple: given N exits on a highway which forms a simple cycle, you are supposed to tell the shortest distance between any pair of exits.
Input Specification:
Each input file contains one test case. For each case, the first line contains an integer N (in [3, 105]), followed by N integer distances D1 D2 … DN, where Di is the distance between the i-th and the (i+1)-st exits, and DN is between the N-th and the 1st exits. All the numbers in a line are separated by a space. The second line gives a positive integer M (<=104), with M lines follow, each contains a pair of exit numbers, provided that the exits are numbered from 1 to N. It is guaranteed that the total round trip distance is no more than 107.
Output Specification:
For each test case, print your results in M lines, each contains the shortest distance between the corresponding given pair of exits.
Sample Input:
5 1 2 4 14 9
3
1 3
2 5
4 1
Sample Output:
3
10
7
思路
一开始用了笨方法,记录各段距离,每次求两点距离时都累加一遍,复杂度O(mn),最后一个测试点果断超时……
后参考了柳神的方法:以0号节点为基准,用dist[i]
表示每个节点到0号的距离,dist[0] = 0
,这样可以直接用dist[i] - dist[j]
求出j到i的距离。由于是环形,距离只有正向反向两种可能,二者和是周长,计算出一边的距离len
,min{len, sum-len}
即为所求,复杂度O(n)。
代码
#include <iostream>
#include <vector>
using namespace std;
int main(){
int n;
cin >> n;
vector<int> dist(n+1);
int sum = 0; //周长
for(int i = 1; i <= n; i++){
int d;
cin >> d;
sum += d;
dist[i] = sum;
}
int m;
cin >> m;
for(int i = 0; i < m; i++){
int s, t;
cin >> s >> t;
if (s > t){
swap(s, t);
}
int len = dist[t-1] - dist[s-1];
cout << min(len, sum - len) << endl;
}
return 0;
}