完成数据结构pta实验7-2 迪杰斯特拉方法实现最短路径
用迪杰斯特拉算法实现有向网的最短路径
输入格式:
第一行输入有向网的顶点和边数，第二行输入各顶点值，用空格间隔，第三行开始输入各条边的 两个点的及边上的权值，用空格间隔。最后一行输入要求路径的两个顶点。
输出格式:
输出最短路径经过的各顶点，中间用-->连接。
输入样例:
在这里给出一组输入。例如：
6 8
0 1 2 3 4 5
0 2 10
0 4 30
0 5 100
1 2 5
2 3 50
3 5 10
4 3 20
4 5 60
0 3
输出样例:
在这里给出相应的输出。例如：
0-->4-->3
点击查看代码
#include<iostream>
#include<vector>
#include<algorithm>
using namespace std;
#define MaxInt 32767
#define MVNum 100
typedef int ArcType;
typedef struct{
int vexs[MVNum];
ArcType arcs[MVNum][MVNum];
int vexnum,arcnum;
}AMGraph;
int LocateVex(AMGraph G,int v){
for(int i = 0;i<G.vexnum;i++){
if(G.vexs[i] == v)
return i;
}
return -1;
}
void CreateUDG(AMGraph &G){
cin>>G.vexnum>>G.arcnum;
for(int i=0;i<G.vexnum;i++){
cin>>G.vexs[i];
}
for(int i=0;i<G.vexnum;i++){
for(int j=0;j<G.vexnum;j++)
G.arcs[i][j]=MaxInt;
}
for(int k = 0;k<G.arcnum;k++){
int v1,v2,w;
cin>>v1>>v2>>w;
int i = LocateVex(G,v1);
int j = LocateVex(G,v2);
if(i != -1&&j != -1){
G.arcs[i][j]=w;
G.arcs[j][i]=G.arcs[i][j];
}
}
}
void ShortestPath_DIJ(AMGraph G,int v0,int v){
int n = G.vexnum;
vector<int> S(n,0);
vector<int> D(n,MaxInt);
vector<int> Path(n,-1);
for(int v=0;v<n;v++){
D[v]=G.arcs[v0][v];
if(D[v]<MaxInt){
Path[v]=v0;
}
}
S[v0]=true;
D[v0]=0;
for(int i=1;i<n;i++){
int min = MaxInt;
int v = -1;
for(int w=0;w<n;w++){
if(!S[w]&&D[w]<min){
v = w;
min = D[w];
}
}
S[v]=true;
for(int w=0;w<n;w++){
if(!S[w]&&(D[v]+G.arcs[v][w]<D[w])){
D[w]=D[v]+G.arcs[v][w];
Path[w]=v;
}
}
}
vector<int>shortestPath;
for(int i=v;i!=-1;i=Path[i]){
shortestPath.push_back(i);
}
reverse(shortestPath.begin(),shortestPath.end());
for(size_t i=0;i<shortestPath.size();i++){
if(i>0)
cout<<"-->";
cout<<G.vexs[shortestPath[i]];
}
}
int main(){
AMGraph G;
CreateUDG(G);
int start,end;
cin>>start>>end;
ShortestPath_DIJ(G,LocateVex(G,start),LocateVex(G,end));
return 0;
}