PAT L2-001. 紧急救援（扩展dijkstra）

参考：http://blog.csdn.net/strokess/article/details/51339933

题目：

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

输入格式：

输入第一行给出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

题意：最短路问题，在路径最短的基础上要求途径的点的权值和最大，并且要打印出路径。同时还要求出满足路径距离最短（不需满足途径点权值和最大）的路径条数。

题解：

扩展的Dijkstra.增加了几个变量作为记录其他几个权值的标准。

int val[MAX],dis[MAX],cou[MAX],vis[MAX];//点权值，起点到该点的最短距离，到该点最优路径的救援队数量和，标记数组

int wayCou[MAX];//到该点最短路径的方法数

int path[MAX]; //最优路径上前驱节点

代码：

#include<iostream>
#include<cstdio>
#include<cstring>
#include<algorithm>
#include<stack>
using namespace std;
const int MAX=500+10;
const int INF=0x3f3f3f3f;
int g[MAX][MAX];
int val[MAX],dis[MAX],cou[MAX],vis[MAX];
int wayCou[MAX];
int path[MAX];
int N,M,S,D;
void dijkstra()
{
for(int i=0;i<N;i++)
{
dis[i]=INF;/*必须是INF，因为多了其他权值，必须自己对自己进行松弛操作，也可以直接最开始进行预处理*/
vis[i]=0;
cou[i]=0;
wayCou[i]=0;
path[i]=-1;
}
dis[S]=0;
vis[S]=1;
cou[S]=val[S];
wayCou[S]=1;
for(int i=1;i<N;i++)
{
int p=S,mmin=INF;
for(int j=0;j<N;j++)
{
if(!vis[j]&&dis[j]<mmin)
{
p=j;
mmin=dis[j];
}
}
vis[p]=1;
for(int j=0;j<N;j++)
{
if(!vis[j])
{
if(dis[j]>dis[p]+g[p][j])
{
dis[j]=dis[p]+g[p][j];
cou[j]=cou[p]+val[j];
wayCou[j]=wayCou[p];
path[j]=p;
}
else if(dis[j]==dis[p]+g[p][j])
{
wayCou[j]+=wayCou[p];
if(cou[p]+val[j]>cou[j])
{
cou[j]=cou[p]+val[j];
path[j]=p;
}
}
}
}
}
}
int main()
{
//freopen("in.txt","r",stdin);
//freopen("out.txt","w",stdout);
while(scanf("%d%d%d%d",&N,&M,&S,&D)!=EOF)
{
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++)
{
g[i][j]=i==j?0:INF;
}
}
while(M--)
{
int s,e,u;
scanf("%d%d%d",&s,&e,&u);
g[s][e]=g[e][s]=min(g[s][e],u);
}
dijkstra();
printf("%d %d\n",wayCou[D],cou[D]);
stack<int> ans;
int p=D;
while(path[p]!=-1)
{
ans.push(path[p]);
p=path[p];
}
while(!ans.empty())
{
printf("%d ",ans.top());
ans.pop();
}
printf("%d\n",D);
}
return 0;
}