5-28 城市间紧急救援 (25分)
作为一个城市的应急救援队伍的负责人,你有一张特殊的全国地图。在地图上显示有多个分散的城市和一些连接城市的快速道路。每个城市的救援队数量和每一条连接两个城市的快速道路长度都标在地图上。当其他城市有紧急求助电话给你的时候,你的任务是带领你的救援队尽快赶往事发地,同时,一路上召集尽可能多的救援队。
输入格式:
输入第一行给出4个正整数NNN、MMM、SSS、DDD,其中NNN(2≤N≤5002\le N\le 5002≤N≤500)是城市的个数,顺便假设城市的编号为0 ~ (N−1)(N-1)(N−1);MMM是快速道路的条数;SSS是出发地的城市编号;DDD是目的地的城市编号。
第二行给出NNN个正整数,其中第iii个数是第iii个城市的救援队的数目,数字间以空格分隔。随后的MMM行中,每行给出一条快速道路的信息,分别是:城市1、城市2、快速道路的长度,中间用空格分开,数字均为整数且不超过500。输入保证救援可行且最优解唯一。
输出格式:
第一行输出最短路径的个数和能够召集的最多的救援队数量。第二行输出从SSS到DDD的路径中经过的城市编号。数字间以空格分隔,输出结尾不能有多余空格。
输入样例:
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
dijskta+dfs。
这道题除了涉及到长度,还涉及到人数,所以在松弛的地方,要多处理一下。即如果路径不一样,就根据路径松弛;如果路径相等,要根据人数松弛。
还有一点就是要记录最短路径的个数。开始时候用了一个全局变量ans,每次发现路径相等了,就自加1.但这种方法只适用于目的点为终点的情况,即把从某一点到其他所有点的最短路径都求了一遍,这时的ans肯定保存了最后那个点的最短路径个数。而题目中的目的地可能是任意点,所以不能直接这样利用全局变量保存。
这里用了类似于动态规划的思想,用一个ans数组。到某一个点最短路径的个数,就等于到它前边那个点的最短路径个数,再加上它本身已经保存了的最短路径个数。
即ans[j] = ans[k] + ans[j]。因为到达j点可能有多种方案,所以它本身保存的最短路径个数是变化的不会恒为1,每从一个点到达j点,就是多了ans[k]种到达j点的方案。
#include <stdio.h>
#include <string.h>
#define INF 999999
int n, m, s, d, vis[505], l[505][505], dis[505], p[505], sum[505], ans[505];
void dfs(int length, int people, int dd)
{
int i;
if(dd == s)
return;
for(i = 0 ; i < n ; i++)
{
if(length - l[dd][i] == dis[i] && people - p[dd] == sum[i])
{
dfs(dis[i], sum[i], i);
printf("%d ", i);
break;
}
}
}
void dij()
{
int i, j, k, min;
vis[s] = 1;
sum[s] = p[s];
ans[s] = 1;
for(i = 0 ; i < n ; i++)
{
dis[i] = l[s][i];
if(s != i && l[s][i] != INF)
{
sum[i] = p[i] + p[s];
ans[i] = 1;//
}
}
for(i = 0 ; i < n-1 ; i++)
{
min = INF;
for(j = 0 ; j < n ; j++)
{
if(!vis[j] && min > dis[j])
{
min = dis[j];
k = j;
}
}
vis[k] = 1;
for(j = 0 ; j < n; j++)
{
if(!vis[j])
{
if(dis[j] > dis[k] + l[k][j])
{
dis[j] = dis[k] + l[k][j];
sum[j] = sum[k] + p[j];
ans[j] = ans[k];//
}
else if(dis[j] == dis[k] + l[k][j])
{
ans[j] += ans[k];//
if(sum[j] < sum[k] + p[j])
{
sum[j] = sum[k] + p[j];
}
}
}
}
}
}
int main()
{
int i, j, x, y, ll;
scanf("%d %d %d %d", &n, &m, &s, &d);
for(i = 0 ; i < n ; i++)
for(j = 0 ; j < n ; j++)
{
l[i][j] = INF;
}
for(i = 0 ; i < n; i++)
{
scanf("%d", &p[i]);
}
for(i = 0 ; i < m ; i++)
{
scanf("%d %d %d", &x, &y, &ll);
l[x][y] = l[y][x] = ll;
}
if(n == 1)
{
printf("1 %d\n1", p[0]);
return 0;
}
dij();
printf("%d %d\n", ans[d], sum[d]);
printf("%d ", s);
dfs(dis[d], sum[d], d);
printf("%d", d);
return 0;
}