PAT甲级 1003 Emergency 单源最短路dijkstra算法解决 C++

1003 Emergency (25分)

这道题目大概的意思是：有N个城市（编号为0~N-1），它们之间一共有M条路，求从编号为C1的城市到C2的最短路径的条数，并求出其最短路径当中能够召集最大队伍的数量。

我在做题时遇到的tips：

1. 可以自己到自己，也就是0到0（测试点1）；
2. 应该为无向图；
3. 最短路径条数的计算。

在牛客网中的测试点中从c1到中间结点的最短路径都只有一条（也就是从c1到w的距离，只有1条），所以刚开始我的代码在牛客没有问题（好像牛客也没有设置自己到自己的测试点），但在PAT却过不了QWQ
好在最后问题解决，还是AK啦～

main函数：

int main(){                         //主函数
#ifdef ONLINE_JUDGE                 //如果有oj系统（在线判定），则忽略文件读入，否则使用文件作为标准输入
#else    
   freopen("1.txt", "r", stdin);   //从1.txt输入数据
#endif 
   int n,m,c1,c2;
   cin>>n>>m>>c1>>c2;
   int *res=new int[n];//存放救援队的数量
   for(int i=0;i<n;i++){
	   cin>>res[i];
   }
   createMaxtrix(m);//构造邻接矩阵
   rescue(n,res,c1,c2);//计算最短路径条数和最大救援队数量
    return 0;                       //返回0，如果不返回0，PAT会报错
}

其它函数：

int a[500][500]={0};
void createMaxtrix(int m){
	for(int i=0;i<m;i++){
		int x,y,d;
		cin>>x>>y>>d;
		a[x][y]=d;
		a[y][x]=d;
	}
}
void rescue(int n,int *res,int c1,int c2){
	int *dist=new int[n];//存放c1到各个点的最短距离
	int *flag=new int[n];//判断是否检查过
	int *res1=new int [n];//存放最大救援队数量
	int *count=new int [n];//存放最短路径数量
	for(int i=0;i<n;i++){//对上述进行初始化
		dist[i]=a[c1][i];
		if(i!=c1&&a[c1][i]==0) dist[i]=2147483647;//如果为a为0，表示没有直接路线
		flag[i]=0;
		res1[i]=res[i];
		count[i]=1;
	}
	flag[c1]=1;
	while(1){
		int w=findMinRoad(n,dist,flag);//寻找没有检查过的点，并且到c1的距离最短的点
		if(w==-1)break;//w为-1表示全部检查过。或者还未全检查但是没有距离最短的点（不连通）
		else{
			flag[w]=1;
			for(int i=0;i<n;i++){//访问w的每个领接点
				if(flag[i]==0){//检查没有检查过的
					if(a[w][i]>0&&a[w][i]+dist[w]<dist[i]){//a>0表示其有直接路径
					dist[i]=dist[w]+a[w][i];
					res1[i]=res[i]+res1[w];
					count[i]=count[w];//注意这里，我就是这里导致一直没有过所有测试点。开始时我将其重新设为1（错误）
                    //count应当是前面c1到w的最短路径条数
					}
					else if(a[w][i]>0&&a[w][i]+dist[w]==dist[i]){
						if(res[i]+res1[w]>res1[i]){
						res1[i]=res[i]+res1[w];
						}
					count[i]=count[i]+count[w];//原来的加上，相等的
					}//else if
				}//if(flag[i]==0)
			}//for
		}//else

	}//while
    int maxres=0;
    if(c1==c2) maxres=res[c1]; //如果为自己到自己
    else maxres=res1[c2]+res[c1];
	cout<<count[c2]<<' '<<maxres<<endl;

}
int findMinRoad(int n,int *dist,int *flag){
	int min=2147483647;int k=-1;
	for(int i=0;i<n;i++){
		if(flag[i]==0){
			if(dist[i]>0&&dist[i]<min){
				k=i;
				min=dist[i];
			}
		}
	}
	return k;
}