模板——最短路径 SPFA

SPFA（最短路径）

题目描述
每组数据第一行包含两个正整数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城市的最短距离；答案
vector <int> ss[maxn];//ss[i]存储的是与第i个城市连接的城市
int n, m;//第一行的N和M
int s, t;//起点、终点
void reset() {//重置、初始化
	memset(vis, 0, sizeof(vis));
	for (int i = 0; i < maxn; i++) {
		for (int j = 0; j < maxn; j++) {
			if (i == j) dp[i][j] = 0;
			else dp[i][j] = inf;
		}
		ss[i].clear();
		len[i] = inf;
	}
}
void SPFA() {//更新len数组
	queue <int> que;
	len[s] = 0;
	que.push(s);
	vis[s] = 1;
	while (!que.empty()) {
		int k = que.front();
		for (int i = 0; i < ss[k].size(); i++) {
			if (len[ss[k][i]] > len[k] + dp[k][ss[k][i]]) {
				len[ss[k][i]] = len[k] + dp[k][ss[k][i]];
				if (vis[ss[k][i]] == 0) {
					que.push(ss[k][i]);
					vis[ss[k][i]] = 1;
				}
			}
		}
		que.pop();
		vis[k] = 0;
	}
}