PAT 1003 Emergency

PAT 1003 Emergency 原题链接

参考博客 链接
题目描述及示例输入输出见上面链接

题目分析：本题给出一个图，图中每个节点有一定数量的救援队，给出起点和终点，希望你输出由起点到终点的最短路径的条数以及沿着最短路径能召唤到的最多的救援队数量。由于最短路径想到用单源最短路径算法Dijkstra算法，但是在本题中需要做一些改变，需要记录最短路径的条数和能召唤到救援队的数量的最大值。先说说Dijkstra算法，简单地说就是不断地选择从到源节点到其他节点的距离的最小值，并用该最小值试着去更新源节点到其他节点的距离值。为了记录最短路径条数应该把Dijkstra中的小于更新距离换为小于等于，并且小于和等于分开处理。除此之外各个数组的初始值设定也应当注意。

#include<iostream>
#include<string.h>
#define MAXN 505
#define inf 9999999
using namespace std;
int res[MAXN];
int map[MAXN][MAXN];

int mark[MAXN];
int dis[MAXN];
int W[MAXN];
int num[MAXN];
int N,to;

void Dijkstra(int from)
{
	memset(mark,0,sizeof(mark));
	W[from] = res[from];
	dis[from] = 0;
	num[from] = 1;
	
	while(1)
	{
		int u,min = inf;
		
		for(int i = 0; i < N; i++)
		{
			if(dis[i] < min && mark[i] == 0)
			{
				min = dis[i];
				u = i;
			}
		}
		if(u == to || min == inf)  break;
		mark[u] = 1;
		
		//用到from最短的点更新其他点
		for(int j = 0; j < N; j++)
		{
			if( dis[u] + map[u][j] < dis[j] && mark[j] == 0)
			{
				dis[j] = dis[u] + map[u][j];
				W[j] = W[u] + res[j];
				num[j] = num[u];
			}
			else if(dis[u] + map[u][j] == dis[j] && mark[j] == 0)
			{
				num[j] += num[u];
				W[j] =max(W[j],W[u]+res[j]);
			}
		}
	}

}
int main()
{
	int M,from,u,v,len;
	cin >> N >> M >> from >> to;
	for(int i = 0; i < N; i++)
		{
			cin >> res[i];
			dis[i] = inf;
			for(int j = 0; j < N; j++)   map[i][j] = inf;
		}

	for(int i = 0; i < M; i++)
		{
			cin >> u >> v >> len;
			map[u][v] = map[v][u] = len;
		}

	Dijkstra(from);
	cout << num[to] << " " << W[to] << endl;

}
/*
5 6 0 4
1 1 1 1 1
0 1 1
0 2 1
0 3 1
1 4 1
2 4 1
3 4 1

*/