PTA1003_Emergency(Dijkstra+dfs)

题目链接：https://pintia.cn/problem-sets/994805342720868352/problems/994805523835109376

解题思路：
首先我们使用Dijkstra算法进行求最短路径，并且使用一个vector数组去记录所有最短路径中的边。然后再使用一个dfs去计数vector数组中所有最短路径的条数，并维护一个点权最大值。

代码：

#include <iostream>
#include <cstdio>
#include <cstring>
#include <vector>

using namespace std;

const int N = 505;

int n, m, s, t;
int g[N][N], w[N];

//找到最短路径的所有边 
int d[N];
bool vis[N];
vector<int> pre[N];
inline void dijkstra(void) {
	memset(vis, false, sizeof vis);
	memset(d, 0x3f, sizeof d);
	d[s] = 0;
	
	for(int i = 1; i <= n; i ++) {
		int t = -1;
		for(int j = 0; j < n; j ++) 
			if(!vis[j])
				if(t == -1 || d[t] > d[j]) t = j;
		
		if(t == -1) return;
		if(d[t] >= 0x3f3f3f3f / 2) return;
		vis[t] = true;
		
		for(int j = 0; j < n; j ++) {
			if(vis[j]) continue;
			if(g[t][j] >= 0x3f3f3f3f / 2) continue;
			if(d[j] > d[t] + g[t][j]) {
				d[j] = d[t] + g[t][j];
				pre[j].clear();
				pre[j].push_back(t);
			} else if(d[j] == d[t] + g[t][j]) {
				pre[j].push_back(t);
			}
		}
	}
}

//维护最短路中的所有的数值
vector<int> path, tmp_path;
int maxx, num;   //记录最大的点权，最短路径的数量 
inline void dfs(int u, int sum) {
	tmp_path.push_back(u);
	sum = sum + w[u];
	
	if(u == s) {
		num ++;
		maxx = max(maxx, sum);
		return;
	}
	
	for(int i = 0; i < pre[u].size(); i ++)
		dfs(pre[u][i], sum);
}
 
int main(void) {
//	freopen("in.txt", "r", stdin);
	scanf("%d%d%d%d", &n, &m, &s, &t);
	for(int i = 0; i < n; i ++) scanf("%d", &w[i]);
	
	memset(g, 0x3f, sizeof g);
	for(int i = 1; i <= m; i ++) {
		int a, b, c; 
		scanf("%d%d%d", &a, &b, &c);
		g[a][b] = g[b][a] = c;
	}

	dijkstra();
	dfs(t, 0);
	printf("%d %d\n", num, maxx);
	
	return 0;
}

注意点：这道题目的点只有400 个，所以使用邻接表去存储图O(n * m)的时间复杂度。
在开始的时候需要初始化g数组。

总结：Dijkstra是用来求单源最短路的算法，但与此同时，还可以去维护一些同时可以计数的数值，一些与最短路径有关的一些变量。如：最短路径的数目，最短路径上点权等等。