团体程序设计天梯赛-练习集 L2-1. 紧急救援 Dijstra单源最短路径拓展应用

L2-1. 紧急救援

时间限制

200 ms

内存限制

65536 kB

代码长度限制

8000 B

判题程序

Standard

作者

陈越

作为一个城市的应急救援队伍的负责人，你有一张特殊的全国地图。在地图上显示有多个分散的城市和一些连接城市的快速道路。每个城市的救援队数量和每一条连接两个城市的快速道路长度都标在地图上。当其他城市有紧急求助电话给你的时候，你的任务是带领你的救援队尽快赶往事发地，同时，一路上召集尽可能多的救援队。

输入格式：

输入第一行给出4个正整数N、M、S、D，其中N（2<=N<=500）是城市的个数，顺便假设城市的编号为0~(N-1)；M是快速道路的条数；S是出发地的城市编号；D是目的地的城市编号。第二行给出N个正整数，其中第i个数是第i个城市的救援队的数目，数字间以空格分隔。随后的M行中，每行给出一条快速道路的信息，分别是：城市1、城市2、快速道路的长度，中间用空格分开，数字均为整数且不超过500。输入保证救援可行且最优解唯一。

输出格式：

第一行输出不同的最短路径的条数和能够召集的最多的救援队数量。第二行输出从S到D的路径中经过的城市编号。数字间以空格分隔，输出首尾不能有多余空格。

输入样例：

4 5 0 3
20 30 40 10
0 1 1
1 3 2
0 3 3
0 2 2
2 3 2

输出样例：

2 60
0 1 3

其实就是dijstra在d的基础上多加几个数组，然后在更新d的时候留意下更新其他几个数组就行了

#include <cstdio>
#include <cstring>
#include <iostream>
#include <cstdlib>

using namespace std;

const int maxn = 500 + 10, INF = 0x3f3f3f3f;
int N, M, S, D;
//val用于存人数，d存到S的最短距离，m存图，cntp存当前点最短路径条数
//num存到当前点最多召集的人数，path存当前点的前驱用于最后输出路径
int val[maxn], d[maxn], m[maxn][maxn], cntp[maxn], num[maxn], path[maxn];
bool vis[maxn];

void dijstra(int s) {
	//初始化
	for (int i = 0; i < N; i++) {
		d[i] = INF;
	}
	for (int i = 0; i < N; i++) {
		path[i] = i;
	}
	memset(cntp, 0, sizeof(cntp));
	memset(num, 0, sizeof(num));
	memset(vis, false, sizeof(vis));
	d[s] = 0;
	cntp[s] = 1;
	num[s] = val[s];
	for (int i = 0; i < N; i++) {
		if (i != s) {
			d[i] = m[s][i];
			if (m[s][i] != INF) {
				cntp[i] = 1;
				num[i] = val[i] + num[s];
				path[i] = s;
			}
		}
	}
	vis[s] = true;
	//初始化完毕

	for (int t = 0; t < N - 1; t++) {
		int minn = INF, index = s;
		for (int i = 0; i < N; i++) {
			if (!vis[i] && minn > d[i]) { //注意!vis[i]
				minn = d[i];
				index = i;
			}
		}
		vis[index] = true;
		for (int i = 0; i < N; i++) {
			if (!vis[i] && m[index][i] < INF) { //debug add !vis[i]
				//不判断vis[i]去重的话会使下一句d[index] + m[index][i] == d[i]即
				//d[index] + 0 == d[i] (i == index时) 为真，重复计算
				if (d[index] + m[index][i] == d[i]) {
					cntp[i] += cntp[index]; //debug 不是简单的cntp[i]++，index之前可能连接着多条路径
					if (num[index] + val[i] > num[i]) {
						num[i] = val[i] + num[index];
						path[i] = index;
					}
				} //此处没判不判断!vis[i]无所谓，是因为即使index == i时，d[index] < d[i]肯定不成立
				else if (d[index] + m[index][i] < d[i]) {
					cntp[i] = cntp[index]; //debug 重新计算最短路径数
					d[i] = d[index] + m[index][i];
					num[i] = val[i] + num[index];
					path[i] = index;
				}
			}
		}
	}
}

int main()
{
	scanf("%d%d%d%d", &N, &M, &S, &D);
	for (int i = 0; i < N; i++) {
		scanf("%d", &val[i]);
	}
	for (int i = 0; i < N; i++) {
		for (int j = 0; j < N; j++) {
			m[i][j] = (i == j ? 0 : INF);
		}
	}
	for (int i = 0; i < M; i++) {
		int u, v, w;
		scanf("%d%d%d", &u, &v, &w);
		m[u][v] = m[v][u] = w;
	}
	dijstra(S);
	printf("%d %d\n", cntp[D], num[D]);
	int pr[maxn], i, len; //用pr相当于一个栈的作用，根据path输出最后结果
	i = D; len = 0;
	while (i != S) {
		pr[len++] = i;
		i = path[i];
	}
	pr[len++] = i;

	for (int j = len - 1; j >= 0; j--) {
		printf(j == len - 1 ? "%d" : " %d", pr[j]);
	}
	puts("");
	return 0;
}