【PAT甲级】1003 Emergency (25分)

解题过程的小记录，如有错误欢迎指出。

难度：三星（题目本身不难，但因为是接触最短路径的第一题，做着有点手生）

题目分析

给出地图（各城市之间的路径长度和各城市的救援队数量），要求给出一个出发城市和一个目标城市，给出这两点之间的最短路径条数，以及在最短路径成立时可以获得的最多的救援队的数量

注意点

Dijkstra中的优化更新板块，当最短路径相等时，一定要更新路径数量（不是单纯的++）

我的解题过程

思路

Dijkstra的模板题
Dijkstra函数中的循环思路：找最小->访问它->优化剩余

bug

1. fill函数太久没写忘了，要注意一维和二维的写法不一样！！！
2. 在优化更新的时候，路径是需要加上前一条路径数的，因为达到前一点有那么多条路径，且当前城市可以被前一点到达，那么当前城市的路径需要加上前一点的路径总数

代码

#include<iostream>

using namespace std;

const int INF = 1000000000;
const int MAXV = 505;
int MIN = INF;
int G[MAXV][MAXV];
bool vis[MAXV] = { false };
int weight[MAXV];//用于存储每个地区的救援队的数量
int w[MAXV];//用于记录到达每个城市能获得的救援队数量
int n, m, c1, c2;
int cnt[MAXV];
int dis[MAXV];

void Dijkstra(int s) {
	fill(dis, dis + MAXV, INF);
	dis[c1] = 0;
	w[c1] = weight[c1];
	cnt[c1] = 1;
	for (int i = 0; i < n; i++) {
		int u = -1;
		MIN = INF;
		for (int j = 0; j < n; j++) {
			if (vis[j] == false && dis[j] < MIN) {
				u = j;
				MIN = dis[j];
			}
		}
		if (u == -1) return;
		vis[u] = true;
		for (int j = 0; j < n; j++) {
			if (G[u][j] != INF&&vis[j] == false) {
				int v = j;
				if (G[u][v] + dis[u] < dis[v]) {
					dis[v] = G[u][v] + dis[u];
					cnt[v] = cnt[u];
					w[v] = w[u] + weight[v];
				}
				else if (G[u][v] + dis[u] == dis[v]) {
					//cnt[v]++; 更新路径数量错误写法
					cnt[v] += cnt[u];//新增可能不是一条哟
					if (w[u] + weight[v] > w[v]) {
						w[v] = w[u] + weight[v];
					}
				}
			}
		}
	}
}

int main()
{
	scanf("%d %d %d %d", &n, &m, &c1, &c2);
	fill(G[0], G[0] + MAXV*MAXV, INF);
	fill(cnt, cnt + MAXV, 0);
	for (int i = 0; i < n; i++) {
		scanf("%d", &weight[i]);
	}
	for (int i = 0; i < m; i++) {
		int u, v, length;
		scanf("%d %d %d", &u, &v, &length);
		G[u][v] = length;
		G[v][u] = length;
	}
	Dijkstra(c1);
	printf("%d %d", cnt[c2], w[c2]);
    return 0;
}

dalao的代码

全部代码因版权原因不放出来，大家可以自行去柳神博客购买或者参考晴神的上机笔记~

借鉴点

本题代码就是参考晴神的代码修正的，柳神的代码把Dijkstra的函数板块写到主函数当中了，被模板建议参考晴神的代码