城市路(Dijkstra)

题目描述

罗老师被邀请参加一个舞会，是在城市n，而罗老师当前所处的城市为1,附近还有很多城市2~n-1，有些城市之间没有直接相连的路，有些城市之间有直接相连的路，这些路都是双向的，当然也可能有多条。

现在给出直接相邻城市的路长度，罗老师想知道从城市1到城市n，最短多少距离。

输入格式

输入n, m，表示n个城市和m条路;

接下来m行，每行a b c， 表示城市a与城市b有长度为c的路。

输出格式

输出1到n的最短路。如果1到达不了n，就输出-1。

输入样例

5

1 2 20

2 3 20

3 4 20

4 5 20

1 5 100

输出样例

90

提示说明

【数据规模和约定】

1≤n≤2000

1≤m≤10000

0≤c≤10000

---

#include <bits/stdc++.h>

using namespace std;

const int oo = 1000000000;
const int M = 2005;
int n, m;
int dis[M], mark[M];
int g[M][M];

void dijkstra(){
	dis[1] = 0;
	mark[1] = 1;
	for(int i = 2; i <= n; i++){
		dis[i] = g[1][i];
		mark[i] = 0; 
	}
	for(int i = 2; i <= n; i++){
		int id;
		int tmp = oo;
		for(int j = 1; j <= n; j++){
			if(mark[j] == 0 && dis[j] < tmp){
				tmp = dis[j];
				id = j;
			}
		}
		mark[id] = 1;
		for(int j = 1; j <= n; j++){
			if(dis[j] > dis[id] + g[id][j]){
				dis[j] = dis[id] + g[id][j];
			}
		}
	}
} 

int main(){
	scanf("%d%d", &n, &m);
	for(int i = 1; i <= n; i++){
		for(int j = 1; j <= n; j++){
			g[i][j] = oo;
		}
		g[i][i] = 0;
	}
	for(int i = 1; i <= m; i++){
		int a, b, c;
		scanf("%d%d%d", &a, &b, &c);
		g[a][b] = g[b][a] = min(c, g[a][b]);
	}
	dijkstra();
	if(dis[n] >= oo){
		dis[n] = -1;
	}
	printf("%d\n", dis[n]);
	return 0;
}