Prim:最小生成树的生成
普里姆算法(Prim算法),图论中的一种算法,可在加权连通图里搜索最小生成树。意即由此算法搜索到的边子集所构成的树中,不但包括了连通图里的所有顶点(英语:Vertex (graph theory)),且其所有边的权值之和亦为最小
- 首先,结构体初始化,邻接矩阵定义,定义一个Cost来观察节点是否被访问
int Graph[110][110];
int Cost[110];
- 初始化图 Init_Graph(int n),将所有节点都标记为未访问(-2),将图的距离定义为无限小
void Init_Graph(int n)
{
for(int i = 1;i <= n; i++)
{
Cost[i] = -2;
}
for(int i = 1;i <= n; i++)
for(int j = 1; j <= n; j++)
{
Graph[i][j] = Graph[j][i] = -2;
}
}
- 定义一个函数:Get_Min(也可以放在Prim_Builder中定义)
功能:取得第一个没有被遍历的节点,得到此节点周围节点权值最小的节点的下标i ,用k进行记录
int Get_Min(int n)
{
//找到第一个没有被遍历的节点
int k = 0;
for(int i = 1; i <= n; i++)
{
if(Cost[i]>-1)
{
k = i;
break;
}
}
//遍历周围节点,找到权值最小的
for(int i = 1; i <= n; i++)
{
if(Cost[i]<Cost[k]&&Cost[i]>-1)
{
k = i;
}
}
return k;
}
- 最小二叉树的建立算法(Prim)
定义 SumCount 用来记录最小生成树所有路径成本之和
然后根据所给的数据把图的邻接矩阵相应的值 赋值给节点的权值
int SumCount = 0;
int k;
Cost[1] = -1;
for(int i = 2; i <= n; i++)
{
Cost[i] = Graph[i][1];
}
接下来是函数主体
- 嵌套一层for循环,循环n-1次,进行最小生成函数的构建需要寻找n-1轮
- 调用Get_Min,用k来接受这个最小节点
- 此时,代表k这个下标的节点已经在最小生成树里了,因此Cost[k]标记为已访问,SumCount += Cost[k]
- 然后,更新表,for循环遍历n-1次,找到最小的,没有被访问的过的节点
int Get_Min(int n)
{
//找到第一个没有被遍历的节点
int k = 0;
for(int i = 1; i <= n; i++)
{
if(Cost[i]>-1)
{
k = i;
break;
}
}
//遍历周围节点,找到权值最小的
for(int i = 1; i <= n; i++)
{
if(Cost[i]<Cost[k]&&Cost[i]>-1)
{
k = i;
}
}
return k;
}
int Prim_BuildTree(int n)
{
int SumCount = 0;
int k;
Cost[1] = -1;
for(int i = 2; i <= n; i++)
{
Cost[i] = Graph[i][1];
}
for(int i = 1; i < n; i++)
{
int k = Get_Min(n);
if(k)
{
SumCount += Cost[k];
Cost[k] = -1; //标记为已访问
//更新表(Cost)
for(int j = 1; j <= n; j++)
{
if(Graph[j][k]<Cost[j]&&Cost[j]>-1||Cost[j]==-2)
{
Cost[j] = Graph[j][k];
}
}
}
else return -1;
}
return SumCount;
}
- 主函数
- 输入顶点个数,路径个数
- 初始化图 Init_Graph(n)
- 将相应路径的值放入图(邻接矩阵)中
- 调用Prim_BuildTree生成最小二叉树即可