dijkstra——迪杰斯特拉(最短路径)
题目描述
每组数据第一行包含两个正整数N和M(0<N<200,0<M<1000),分别代表现有城镇的数目和已修建的道路的数目。城镇分别以0~N-1编号。
接下来是M行道路信息。每一行有三个整数A,B,X(0<=A,B<N,A!=B,0<X<10000),表示城镇A和城镇B之间有一条长度为X的双向道路。
再接下一行有两个整数S,T(0<=S,T<N),分别代表起点和终点。
计算出要从起点到终点,最短需要行走多少距离?
const int maxn = 205;//N的上限
const int inf = 1e9; //数据的上限
int dp[maxn][maxn];//dp[i][j]:第i个城市到第j个城市的距离
int vis[maxn];//vis[i]:标记第i个城市是否遍历过
int len[maxn];//len[i]:第i个城市到s城市的最短距离;答案
int n, m;//第一行的N和M
int s, t;//起点、终点
void reset(void) {//清空数据、初始化
memset(vis, 0, sizeof(vis));
for (int i = 0; i < n; i++) {
for (int j = 0; j < n; j++) {
dp[i][j] = i == j ? 0 : inf;
}
}
}
void dijkstra() {//更新len数组的数据
for (int i = 0; i < n; i++) {
len[i] = dp[s][i];
}
vis[s] = 1;
while (1) {
int minn = inf;
int k = -1;
for (int i = 0; i < n; i++) {
if (vis[i] == 0 && minn > len[i]) {
minn = len[i];
k = i;
}
}
if (k == -1) {
break;
}
vis[k] = 1;
for (int i = 0; i < n; i++) {
if (vis[i] == 0 && len[i] > len[k] + dp[k][i]) {
len[i] = len[k] + dp[k][i];
}
}
}
}