PAT 1003 Emergency
题目大意:有n个城市和m条路,每个城市有救援小组,所有边边权已知,所有点点权已知(救援小组数量),给定起点和终点,求起点到终点的最短路径之和以及最短路径上的最大点权之和(沿途能召集到的救援小组数量之和)
分析:重点是要求出从单个节点出发到图中其他各节点的最短路径,然后才能求出相应数据。用一次Dijkstra算法,dis[i]标识从出发点到i节点的最短路径长度,num[i]表示到i节点的最短路径条数,w[i]表示到i节点的点权之和
// 1003 Emergency.cpp : 此文件包含 "main" 函数。程序执行将在此处开始并结束。
//
#include <iostream>
#include <algorithm>
using namespace std;
int n, m, c1, c2; //城市数量n,道路数量m,出发城市c1,目标城市c2
//e保存边权,weight保存点权,dis[i]保存到城市i的最短路径长度,num[i]保存到城市i的最短路径条数,w[i]保存到城市i点权之和
int e[510][510], weight[510], dis[510], num[510], w[510];
bool visit[510]; //保存城市i是否已被访问过
const int inf = 99999999;
int main()
{
cin >> n >> m >> c1 >> c2;
for (int i = 0; i < n; i++) cin >> weight[i];
fill(e[0], e[0] + 510 * 510, inf); //边权初始化为inf
fill(dis, dis + 510, inf); //最短距离初始化为inf
int a, b, c;
for (int i = 0; i < m; i++) {
cin >> a >> b >> c;
e[a][b] = e[b][a] = c; //初始化边权矩阵
}
dis[c1] = 0; //到自己的路径长度为0
w[c1] = weight[c1]; //到自己的点权和为自己的点权
num[c1] = 1; //到自己的最短路径数为1
for (int i = 0; i < n; i++) {
int u = -1, minn = inf; //u保存当前距初始节点距离最短的节点,minn保存他们之间的距离
for (int j = 0; j < n; j++) {
if (visit[j] == false && dis[j] < minn) { //已访问过的节点不再访问
u = j;
minn = dis[j];
}
}
if (u == -1) break; //所有节点都已访问过,退出循环
visit[u] = true; //将这个节点的状态更新为已访问过
for (int v = 0; v < n; v++) { //遍历
if (dis[u] + e[u][v] < dis[v]) {
dis[v] = dis[u] + e[u][v]; //从u出发到v如果比当前已知到v的最短距离小,更新最短距离
num[v] = num[u]; //到v的最短路径数实际上等于到u的最短路径数
w[v] = w[u] + weight[v]; //更新点权之和
}
else if (dis[u] + e[u][v] == dis[v]) {
num[v] += num[u]; //到v的最短路径还可以通过u走通
if (w[v] < w[u] + weight[v]) {
w[v] = w[u] + weight[v]; //更新点权之和
}
}
}
}
cout << num[c2] << " " << w[c2];
return 0;
}
fill函数,用于填充一维数组、容器、二维数组等,包含在< algorithm>中
fill(e[0], e[0] + 510 * 510, inf); //边权初始化为inf
fill(dis, dis + 510, inf); //最短距离初始化为inf
fill(first, last, val),它的赋值范围为[first, last),因此不包括last,赋的值为val
详情参考C++ 中 fill() 的使用_LawsonAbs’s Spiritual Home-CSDN博客
大数组一定要作为全局变量保存,而不是写在main函数中,否则容易发生栈溢出的情况(段错误),因为大数组在main函数里面将被存储在栈中,而栈是在进程创建时初始化的,有固定的大小,一般为几十KB,所以太大的数组会耗光栈空间。
而全局变量占用的是堆空间,堆空间中的内存是按需分配,自由增长的,可以非常大,例如在32位系统中可以达到4GB,这样可以避免栈溢出
Dijkstra算法:单源最短路径算法
从一个顶点到其余各顶点的最短路径算法,解决的是有权图中的最短路径问题。
采用贪心算法策略,每次遍历到起始点距离最近且未访问过的顶点的邻接节点,直到扩展到终点为止
详细算法看书