PTA Advanced 1003 Emergency C++

稍微变动Dijkstra算法即可

#include <iostream>
using namespace std;

const int MAXINT=0x7fffffff; 

int matrix[500][500];// 邻接矩阵（这个二维数组比较大，放在函数栈里面不合理，浪费栈空间，且栈也容不下这么大的空间，所以一定要放全局区或者堆区里面） 
int numTeams[500];// 结点权值 
int roads[500*499/2][3];// 边：边起点、边终点、边权值

//  改良dj算法，根据邻接矩阵算出从起点到终点的最短路径个数
// 以及最短路径中经过的结点的权值和（包含起点和终点）
int pathLens[500];// 从起点到所有结点的路径的长度记录数组 
bool visited[500];// 记录inCity到第i个结点的最短路径是否被确定下来 
int spathNums[500];// 最短路径条数数组 
int maxMetricSums[500];// 最短路径上结点的权值和的最大值 
void djImproved(int matrix[500][500],int numNode,int inCity){	
	// 初始化路径长度数组、访问数组、最短路径条数数组和最大权值和数组 
	for(int i=0;i<numNode;i++){
		pathLens[i]=matrix[inCity][i];
		visited[i]=false; 
		spathNums[i]=0;
		maxMetricSums[i]=0; 
	} 
	pathLens[inCity]=0;// 自己到自己的路径长度为0
	spathNums[inCity]=1;
	maxMetricSums[inCity]=numTeams[inCity];
	
	// 进行n次选择，每次都会选中一个结点，确定inCity到其的最短路径长度  
	for(int i=0;i<numNode;i++){
		// 每次都是选中路径长度数组中的最小值
		int minNode=inCity;// 最小node的编号
		int minPath=MAXINT;// 最短的路径值 
		for(int j=0;j<numNode;j++){
			if(visited[j]==false&&pathLens[j]<minPath){
				minPath=pathLens[j];
				minNode=j;
			}
		} 	
		
		visited[minNode]=true;// 到该节点的最短路径确定了下来
		
		// 检查经过该节点到达其他节点的路径是否更短
		for(int j=0;j<numNode;j++){
			if(visited[j]==false&&matrix[minNode][j]!=MAXINT){
				if(pathLens[minNode]+matrix[minNode][j]<pathLens[j]){// 找到了更短的路径 
					pathLens[j]=pathLens[minNode]+matrix[minNode][j];
					spathNums[j]=spathNums[minNode];
					maxMetricSums[j]=maxMetricSums[minNode]+numTeams[j];
				}else if(pathLens[minNode]+matrix[minNode][j]==pathLens[j]){// 又找到了一种经过minNode类型的最短路径（可能包含多条路径） 
					// 易错点：这里不是简单的+1，而是加上到达minNode的所有最短路径的条数 
					spathNums[j]+=spathNums[minNode]; 
					if(maxMetricSums[minNode]+numTeams[j]>maxMetricSums[j])
						maxMetricSums[j]=maxMetricSums[minNode]+numTeams[j];
				}					
			}
		} 
	} 
	return;	
}

int main() {
    int numCity;// 结点数
	int numRoad;// 边数
	int inCity;// 起点
	int outCity;// 终点
	cin>>numCity>>numRoad>>inCity>>outCity;
	
	// 初始化结点权值 
	for(int i=0;i<numCity;i++){
		cin>>numTeams[i];
	}
 
	// 初始化邻接矩阵 
	for(int i=0;i<numCity;i++){
		for(int j=0;j<numCity;j++){
			matrix[i][j]=MAXINT;
		}
		matrix[i][i]=0;
	} 
	
	// 初始化边 
	for(int i=0;i<numRoad;i++){
		for(int j=0;j<3;j++){
			cin>>roads[i][j];
		}
		// 易错点：构造邻接矩阵时，如果是无向图，需要构造成对称矩阵 
		matrix[roads[i][0]][roads[i][1]]=roads[i][2];// 构造邻接矩阵 
		matrix[roads[i][1]][roads[i][0]]=roads[i][2];
	} 
	
	djImproved(matrix,numCity,inCity);// 改良dj算法，根据邻接矩阵算出从起点到终点的最短路径个数以及最短路径中经过的结点的权值和（包含起点和终点） 
    
	cout<<spathNums[outCity]<<' '<<maxMetricSums[outCity];
	return 0;
}