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
Dijkstra算法的题尽量使用二维动态数组,普通的二维数组很多情况下都会内存超限或者运行超时
#include<iostream>
#include<vector>
using namespace std;
#define INF 10001
struct Node{
int v,l;//一条路的尾结点,该路的长度
};
int n,e,start,stop;
vector<Node> v[1001];
//使用二维动态数组 ,其实二维动态数组和普通数组的唯一区别就在于定义不一样,其他的使用方法和意义都是一样的 ,所以千万不要觉得麻烦
int dist[1001];
int path[1001];
int visit[1001];
void Dijkstra(int start){
fill(dist,dist + 1001,INF);
dist[start] = 0;
for(int i = 0;i < n;i ++){
int u = -1,minx = INF;
for(int j = 0;j < n;j ++){
if(visit[j] == 0 && dist[j] < minx){
minx = dist[j];
u = j;
}
}
if(u == -1) break;
visit[u] = 1;
for (int j = 0; j < v[u].size(); j++) {
int x = v[u][j].v;
if (visit[x] == 0 && dist[u] + v[u][j].l < dist[x]) {
dist[x] = dist[u] + v[u][j].l;
path[x] = u;
}
}
}
}
int main(){
cin >> n >> e;
for (int i = 0; i < e / 2; i++) {
int a, b, c;
cin >> a >> b >> c;
//这两行就相当于用二维数组的edge[a][b] = edge[b][a] = c;但是很多情况下用二维数组都会运行超时,要不然就内存超限
v[a].push_back(Node{b, c});
v[b].push_back(Node{a, c});
}
cin >> start >> stop;
if (start == stop) {//如果起点和终点是同一个结点
printf("%d-->%d:0",start,start);
return 0;
}
Dijkstra(start);
vector<int> ve;
int x = stop;
while (x != start) {
ve.push_back(x);
x = path[x];
}
cout << start;
for (int i = ve.size() - 1; i >= 0; i--)
cout << "-->" << ve[i];//动态数组的输出也不一定需要迭代器
cout << ":" << dist[stop];
}