前两天做了一下医院设置问题,中间卡壳的地方也来查了一番,发现大佬们写的对新手小菜实在不太友好,所以把自己的代码发上来,希望能让大家更容易看懂。
问题描述:n个村庄之间的交通图用无向加权图表示,图中的无向边(vi,vj)表示第i个村庄和第j个村庄之间有道路,边上的权表示这条道路的长度。现在要从这n个村庄中选择一个村庄建一所医院,问这所医院应建在哪个村庄,才能使离医院最远的村庄到医院最近。
#include<stdio.h>
#include<stdlib.h>
int main(){
int n1;
scanf("%d",&n1);//n1个村庄
int *a=(int*)malloc(n1*sizeof(int));
bool *visited=(bool*)malloc(n1*sizeof(bool));
for(int i=0;i<n1;i++){
scanf("%d",&a[i]);//给出村庄序号(为了方便后续操作,选择用数字表示村庄)
}
int **e=(int**)malloc(n1*sizeof(int*));//邻接矩阵,存储村庄间的边
for(int i=0;i<n1;i++){
e[i]=(int*)malloc(n1*sizeof(int));
}
for(int i=0;i<n1;i++){
for(int j=0;j<n1;j++){
if(i!=j) e[i][j]=1000;
else e[i][j]=0;
}
}
int n2,t1,t2,length;
scanf("%d",&n2); //n2条道路
for(int i=0;i<n2;i++){
scanf("%d %d %d",&t1,&t2,&length); //村庄间道路长
e[t2-1][t1-1]=e[t1-1][t2-1]=length; //无向图邻接矩阵对称特性
}
//Dijkstra算法
int **distance=(int**)malloc(n1*sizeof(int*)); //二维数组存储每个村庄到其他村庄的路途长短
for(int i=0;i<n1;i++){
distance[i]=(int*)malloc(n1*sizeof(int));
}
for(int i=0;i<n1;i++){
for(int j=0;j<n1;j++){
distance[i][j]=e[i][j];//初始距离与邻接矩阵相同
}
}
int sum,min;
int closest; //用来标记与某顶点距离最近的顶点的下标
for(int i=0;i<n1;i++){
min=1000;
for(int j=0;j<n1;j++){
visited[j]=false;
}
for(int j=0;j<n1;j++){
if(i!=j&&!visited[j]&&distance[i][j]<min){
min=distance[i][j];
closest=j; //找到距离最近的顶点的下标
}
}
visited[closest]=true;
for(int j=0;j<n1;j++){
if(!visited[j]&&e[closest][j]<1000){
sum=distance[i][closest]+e[closest][j];
if(sum<distance[i][j])
distance[i][j]=sum; //计算不能直接到达的两村庄间的距离
}
}
}
int *farthest=(int*)malloc(n1*sizeof(int));//最远距离
int tempfar; //暂时存储一个村庄到其他村庄的最远距离
for(int i=0;i<n1;i++){
tempfar=0;
for(int j=0;j<n1;j++){
if(tempfar<distance[i][j])
tempfar=distance[i][j];
}
farthest[i]=tempfar;
}
min=1000;
int answer; //答案村庄的下标
for(int i=0;i<n1;i++){
if(min>farthest[i]){
min=farthest[i];
answer=i; //经过比较,找到题目所求的村庄的下标
}
}
printf("建在%d号村\n",a[answer]);
for(int i=0;i<n1;i++){
free(e[i]);free(distance[i]);
}
free(a);free(visited);free(farthest);
return 0;
}
如果还有看不懂的地方,欢迎留言。