PAT甲级 1046 Shortest Distance（前缀和）

Liu_Xingyun

于 2024-07-15 01:45:53 发布

阅读量129

点赞数 3

文章标签： pat考试

本文链接：https://blog.csdn.net/Liu_Xingyun/article/details/140426465

版权

原题链接：

https://pintia.cn/problem-sets/994805342720868352/exam/problems/type/7?problemSetProblemId=994805435700199424&page=0

注意点：

直接遍历计算一对出口之间的最短距离会测试点3会超时。
使用前缀和，提前处理数据（ $pre[i]$ 代表 $D_{1}+D_{2}+...+D_{i}$ ），简化之后计算出口之间最短距离的过程。

代码：

#include <iostream>
#include <algorithm>
#include <math.h>
using namespace std;

#define endl '\n'

int N;//出口数量
int D[100010];
int M;
int pre[100010];//前缀和

signed main() {
	ios::sync_with_stdio(0), cin.tie(0), cout.tie(0);

	cin >> N;
	for (int i = 1; i <= N; i++) {
		cin >> D[i];
		pre[i] = pre[i - 1] + D[i];
	}
	cin >> M;
	for (int i = 0; i < M; i++) {
		int a, b;
		cin >> a >> b;
		if (a > b) {
			swap(a, b);
		}
		int distance = pre[b-1] - pre[a-1];
		distance = min(distance, pre[N]-distance);
		cout << distance << endl;
	}

	return 0;
}