【前缀和计算】PAT A1046 Shortest Distance

实现

码前思考

1. 首先注意到运行时间的限制为200ms，对于一般的OJ系统，1s（即1000ms）中能计算$10^7$~$10^8$次。所以我们不能使用暴力的求解，因为暴力解法的时间复杂度为$MN$，大约有$10^9$方了，所以得想其他方法。
2. 既然暴力不行，那么只能另辟蹊径，转换思维，首先$M$是不可能变的，那么只有把$N$变成1，做法就是选取参考系，计算相对距离。

代码实现

//首先，一定不能暴力求解，不然会超时
//时间限制为200ms，1s允许的操作为10^7~10^8操作
#include "bits/stdc++.h"
using namespace std;
//最大数组长度 
const int maxn = 1e5+10;

//表示整个环的长度 
int total; 

//表示顺时针方向，起点1到这个结点的距离
int dis[maxn]; 

//顶点数
int n;

//对数
int m; 
 
int main(){
	scanf("%d",&n);
	
	//题目保证不会超限int的 
	total = 0;
	
	for(int i=0;i<n;i++){
		int d;
		scanf("%d",&d);
		total = total + d;
		//同时赋值
		dis[(i+2)%(n+1)] = total; 
	}
	
	//设置1为0 
	dis[1] = 0; 
	
	scanf("%d",&m);
	//下面开始读入数据
	for(int i=0;i<m;i++){
		int a;
		int b;
		scanf("%d %d",&a,&b);
		//得到那个顶点是大的顶点
		if(a < b){
			swap(a,b);
		}
		//a表示序号大的点，b表示序号小的点
		int d1 = dis[a] - dis[b];
		int d2 = total - d1;
		
		//取两者的最小值 
		printf("%d\n",min(d1,d2)); 
	}
		
	return 0;
}

码后反思

1. 要认真分析时间复杂度！

二刷代码

#include <iostream>
#include <algorithm>
using namespace std;
const int maxn = 1e5+10;

int sum=0;
int dis[maxn];
int n;
int m;

int main(){
	//freopen("input.txt","r",stdin);
	scanf("%d",&n);
	dis[1]=0;
	int d;
	for(int i=2;i<=n;i++){
		scanf("%d",&d);
		sum+=d;
		dis[i]=sum;
	}
	scanf("%d",&d);
	sum+=d;
	
	scanf("%d",&m);
	for(int i=0;i<m;i++){
		//cout<<"i: "<<i<<endl;
		int u,v;
		scanf("%d %d",&u,&v);
		int d1 = abs(dis[u]-dis[v]);
		int d2 = sum-d1;
		printf("%d\n",d1>d2?d2:d1);	
	}
	//cout<<"e";
	return 0;
}