图数组存储的构造函数
依然保持着自己的习惯,每个顶点都从1开始计算,不从0,方便输入。
比之前的构造函数增加了边权,如果没有赋权则为INF(无穷大)。
MGraph(int a[], int n, int e)
{
arcNum = e;
vertexNum = n;
for (int i = 1; i <= vertexNum; ++i)
{
vertex[i] = a[i];
}
for (int i = 1; i <= vertexNum; ++i)
{
for (int j = 1; j <= vertexNum; ++j)
{
arc[i][j] = INF;
}
}
for (int i = 1; i <= arcNum; ++i)
{
int a, b, v;
cin >> a >> b >> v;
arc[a][b] = arc[b][a] = v;
}
}
Prime算法思想
我们把一个整图定义为V,顶点(第一个开始的点)所在的集合成为U,这样还没有连接到最小生成树中的顶点就属于集合(U-V)中。
这里还需要定义两个额外的辅助数组:
lowcost[n]:用来表示(V-U)中的各个顶点与集合(U)中的顶点最短边的权值,如果lowcost[v]=0,则表示顶点 v 已经加入到最小生成树之中。
adjvex[n]:用来保存集合(U-V)中的顶点与集合(U)中连接的最小边的顶点。这个顶点是 U 中的。
代码思想:
1.首先初始化 两个辅助数组,lowcost[i]=arc[1][i] ,adjdex[i]=1 。 表示以1为最开始的顶点,与1顶点相连接的边的边权就等于arc数组的值。 adjdex为1表示图V中各个顶点的连接都与1顶点相连接。
2.给lowcost[1]=0,设置起点顶点到自身距离为0 。
3.执行n-1次循环。
3.1. 在lowcost[n]中选取最小边,lowcost[k],去找到相应的顶点k。
3.2. 输出顶点k的顶点信息和与起点顶点对应的权值。
3.3. 将顶点k加入的集合U之中,(lowcost[k]=0)
3.4. 调整lowcost和adjvex数组
Prime算法实现:
void Prime()
{
for (int i = 2; i <= vertexNum; ++i)
{
lowcost[i] = arc[1][i];
adjvex[i] = 1;
}
lowcost[1] = 0;
for (int i = 2; i <= vertexNum; ++i)
{
int k = MinEdge(lowcost, vertexNum);
cout << k << " " << adjvex[k] << " " << lowcost[k] << endl;
lowcost[k] = 0;
for (int j = 1; j <= vertexNum; ++j)
{
if (arc[k][j] < lowcost[j])
{
lowcost[j] = arc[k][j];
adjvex[j] = k;
}
}
}
}