最短路径问题(CODEVS 2602)

用Dijkstra算法解决，通过边松弛实现

题目链接

#include<iostream>
#include<cmath>
#include<iomanip>
using namespace std;
int n,m;
double e[105][105];//存储两边之间的距离 
int inf=999999999;
double dis[105];
int book[105];
struct point{
	int x;
	int y;
}arr_node[105];
//算两点之间距离 
double findD(point n1,point n2){
	return sqrt(fabs(n1.x-n2.x)*fabs(n1.x-n2.x)+fabs(n1.y-n2.y)*fabs(n1.y-n2.y));
}
//初始化数组e 
void initE(){
	for(int i=1;i<=n;i++){
		for(int j=1;j<=n;j++){
			if(i==j){
				e[i][j]==0;
			}else{
				e[i][j]=inf;
			}
		}
	}
}
//初始化数组dis 
void initDis(int start,int n){
	for(int i=1;i<=n;i++){
		dis[i]=e[start][i];
	}
}
//初始化标记数组book 
void initBook(int n){
	for(int i=1;i<=n;i++){
		book[i]=0;
	}	
}
//算法核心 
void Dijkstra(int start,int n){
	for(int i=1;i<=n-1;i++){
		int min=inf;
		int u;
		for(int j=1;j<=n;j++){
			if(book[j]==0&&dis[j]<min){
				u=j;
			}
		}	
		book[u]=1;
		for(int v=1;v<=n;v++){
			if(e[u][v]<inf){
				if(dis[v]>dis[u]+e[u][v]){
					dis[v]=dis[u]+e[u][v];
				}
			}
		}
	}
}
int main(){
	cin>>n;
	for(int i=1;i<=n;i++){
		//录入每个点 
		cin>>arr_node[i].x;
		cin>>arr_node[i].y;
	}	
	initE();
	cin>>m;
	int n1,n2;
	for(int i=1;i<=m;i++){
		cin>>n1>>n2;
		//计算两点距离，注意这里是无向图 
		e[n1][n2]=findD(arr_node[n1],arr_node[n2]);
		e[n2][n1]=findD(arr_node[n1],arr_node[n2]);
	}
	int start,end;
	cin>>start>>end;
	initDis(start,n);
	Dijkstra(start,n);
	cout<<fixed<<setprecision(2)<<dis[end];
	return 0;
}