L2-001. 緊急救援Dijkstra算法（單源最短路徑）

L2-001. 緊急救援

時間限制

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

#include<stdio.h>
#include<string.h>

int u,v,w;
int n,m,s,d;
int matr[600][600];
int path[600];
int dist[600];
int pv[600];
int vis[600];
int help[600];

void DijkstraPath(int s)
{
	int i,j,k;
	int vis[600];
	for(i = 0;i<n;i++)
	{
		if(matr[s][i]>0&&i!=s)
		{
			dist[i] = matr[s][i];
			path[i] = s;
			pv[i] = help[s]+help[i];
		}
		else
		{
			dist[i] = 600;
			path[i] = -1;
		}
		vis[i] = 0;
		path[s] = s;
		dist[s] = 0;		
		
	}	
	vis[s] = 1;
	for(i = 1;i<n;i++)
	{
		int min = 600;
		int uu;
		for(j = 0;j<n;j++)
		{
			if(vis[j]==0&&dist[j]<min)
			{
				min = dist[j];
				uu = j;
			}
		}
		vis[uu] = 1;
		for(k = 0;k < n; k++)
		{
			if(vis[k] == 0&&matr[uu][k]>0&&min+matr[uu][k]<dist[k])
			{
				dist[k] = min + matr[uu][k];
				path[k] = uu;
			}
			else if(vis[k] == 0&&matr[uu][k]>0
				&&min+matr[uu][k]==dist[k]&&pv[j]<pv[uu]+help[j])
				{
					pv[j]=pv[uu]+help[j];
					path[k] = uu;
				}
		}
	}
}


int main()
{

	scanf("%d%d%d%d",&n,&m,&s,&d);
	
	memset(matr,0,sizeof(matr));
	for(int i = 0; i < n;i++)
	{
		scanf("%d",&help[i]);
	}
	for(int i = 0; i < m;i++)
	{
		scanf("%d",&u);
		scanf("%d",&v);
		scanf("%d",&w);
		matr[u][v] = w;
	}
	DijkstraPath(s);
	int way[600];
	int l=0;
	int allhelp = 0;	
	while(1)
	{
		
		way[l++] = d;
		allhelp +=help[d];
		d = path[d];
		
		if(d == s )
		{
			way[l++] = d;
			allhelp +=help[d];
			d = path[d];
			break;
		}
	}
	printf("%d %d\n",--l,allhelp);
	for(int i = l;i>=0;i--)
	{
		
		printf("%d",way[i]);
		if(i)
		printf(" ");
	}
	
	
 }