1003 Emergency (25分)
这道题目大概的意思是:有N个城市(编号为0~N-1),它们之间一共有M条路,求从编号为C1的城市到C2的最短路径的条数,并求出其最短路径当中能够召集最大队伍的数量。
我在做题时遇到的tips:
- 可以自己到自己,也就是0到0(测试点1);
- 应该为无向图;
- 最短路径条数的计算。
在牛客网中的测试点中从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;
}