输 入 格 式 : {\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}结果:}
结果: