2021天梯赛训练-10——7-8 最短路径之Dijkstra (10 分)

本题目要求通过读入无向网的边的信息（省略了各顶点的信息，仅用顶点编号来表示），构造图，并利用Dijkstra算法，求出指定源点到其它各点的最短路径。

输入样例:
第一行，两个整数，顶点数vN和边数eN。 以后若干行，是相关边的信息，无向图的边是对称的，只输入一半的边（小编号到大编号的，间以空格），最后两行各一个整数，前一个指定源点，后一个指定的查询的终到点。 （注意，示例中34条边，只输入了17条边的信息）

10 34
0 1 2
0 3 5
1 2 5
1 3 2
2 4 8
2 5 4
3 5 4
3 6 2
4 7 5
4 5 2
5 6 3
5 7 9
5 8 7
6 8 7
7 8 3
7 9 4
8 9 8
0
8

输出样例:
在一行中输出从源点到指定终点的短路径及代价，注意：所有符号均为西文符号。

0-->1-->3-->6-->8:13

思路
最后一个测试点过不去，大概是考虑源点终点相同时，应输出0–>0:0
代码

#include <bits/stdc++.h>
using namespace std;
#define MAX 99999
int xx[10001][10001],cost[10001],pre[10001],visit[10001];
void dijkstra(int start,int end,int n){
	memset(visit,0,sizeof(visit));
	memset(pre,-1,sizeof(pre));
	for(int i=0;i<=n;i++){
		cost[i]=xx[start][i];
		pre[i]=start;
	}
	pre[start]=-1;
	visit[start]=1;
	cost[start]=0;
	for(int i=0;i<n-1;i++){
		int minpos=n;
		for(int j=0;j<n;j++){
			if(visit[j]==0&&cost[j]<cost[minpos]) minpos=j;
		}
		if(minpos==n) break;
		visit[minpos]=1;
		for(int j=0;j<n;j++){
			if(visit[j]==0&&cost[j]>cost[minpos]+xx[minpos][j]){
				cost[j]=cost[minpos]+xx[minpos][j];
				pre[j]=minpos;
			}
		}
	}
}
int main(){
	int v,e,a,b,c,start,end;
	cin>>v>>e;
	for(int i=0;i<=v;i++){
		for(int j=0;j<=v;j++){
			xx[i][j]=xx[j][i]=MAX;
		}
	}
	for(int i=0;i<e/2;i++){
		cin>>a>>b>>c;
		xx[a][b]=xx[b][a]=c;
	}
	cin>>start>>end;
	dijkstra(start,end,v);
	vector<int> v1;
	for(int i=pre[end];i!=-1;i=pre[i]){
		v1.push_back(i);
		
	}
	cout<<start<<"-->";
	for(int i=v1.size()-2;i>=0;i--){
		cout<<v1[i]<<"-->";
	}
	cout<<end<<":"<<cost[end];
	return 0;
}