输 入 格 式 : {\color{Violet}输入格式:} 输入格式:
∙ \bullet ∙ 第一行三个数n、m、start,分别代表城市个数、道路条数、出发点(起始点)。
∙ \bullet ∙ 接下来m行每行三个数a、b、c,分别表示城市a、城市b、城市a和b之间的距离。
输 出 格 式 : {\color{Violet}输出格式:} 输出格式:
∙
\bullet
∙ 从起始点开始,每行输出一条选中的路径,格式为:a--b:dis
,表示从a到b,距离为dis。
∙ \bullet ∙ 最后一行输出总的生成代价sum_dis。
定 义 存 储 数 组 : {\color{Violet}定义存储数组:} 定义存储数组:
∙
\bullet
∙ int road[max_city][max_city]
,存储城市与城市之间的距离矩阵 ,初始化为无穷大。
∙
\bullet
∙ int dist[max_city]
,记录距离该城市最近的一个城市距离,初始化为无穷大。
∙
\bullet
∙ int jilu[max_city]
,存储距离该城市最近的城市编号,即jilu[s]=v
表示距离城市s最近的城市是v。
∙
\bullet
∙ bool visited[max_city]
,记录是否已经被访问,初始化为false。
代 码 : {\color{Violet}代码:} 代码:
#include<iostream> using namespace std; const int INF=99999999; int road[1005][1005]; int dist[1005]; int jilu[1005]; bool visited[1005]; void prim(int n,int start) { dist[start]=0; int res=0; for(int i=0;i<n;i++) { int s=-1; int min=INF; for(int j=0;j<n;j++) { if(visited[j]==false && dist[j]<min) { s=j; min=dist[j]; } } if(s==-1) break; else { visited[s]=true; } res+=dist[s]; //输出最小生成树的边 if(s!=start) cout<<jilu[s]<<"--"<<s<<" : "<<dist[s]<<endl; for(int v=0;v<n;v++) { if(visited[v]==false && road[s][v]!=INF && road[s][v]<dist[v]) { dist[v]=road[s][v]; jilu[v]=s; } } } //输出总的代价 cout<<"sum_dis = "<<res; } int main() { int n,m,start; cin>>n>>m>>start; //初始化,都定义为无穷大 fill(road[0],road[0]+1005*1005,INF); fill(dist,dist+1005,INF); //输入起点(a)、终点(b)、距离(c),并存入距离矩阵中 for(int i=0;i<m;i++) { int a,b,c; cin>>a>>b>>c; road[a][b]=c; road[b][a]=c; } prim(n,start); system("pause"); return 0; } /* 测试用例 5 6 0 0 1 1 0 2 2 0 3 1 1 2 3 2 4 1 3 4 1 */
结
果
:
{\color{Violet}结果:}
结果: