python中的图结构——最小生成树

最小生成树

一个有 n 个结点的连通图的生成树是原图的极小连通子图，且包含原图中的所有 n 个结点，并且有保持图连通的最少的边。
完成构造网的最小生成树必须解决下面两个问题：

（1）尽可能选取权值小的边，但不能构成回路；

（2）选取n－1条恰当的边以连通n个顶点；

图的建立

主体部分见前一篇：https://blog.csdn.net/nanashi_F/article/details/92219999
为了能获取权值得到最小生成树，在AdGraph中添加了一个函数

获取两结点之间边的权值 GetWeight（begin,end）

    def GetWeight(self,begin,end):      #获取两结点之间边的权值
        wei=10000
        p=self.vertexList[begin].firstNode
        if p!=None:
            if p.Index==end:
                wei=p.Weight
            else:
                while p.Next!=None:
                    p=p.Next
                    if p.Index==end:
                        wei=p.Weight
        return wei

Prim算法

基本思路

所有节点分成两个group，一个为已经选取的Select，一个为未选的Candidate。
首先将已给的一个节点加入到Select，然后遍历头节点在Select、尾节点在Candidate的边，选取符合这个条件的边里面权重最小的边，加入到最小生成树，选出的边的尾节点加入到Select，并从Candidate删除。重复此操作直到Candidate中没有节点。
需要注意的是，循环开始前要保证所有节点都有边连接，否则可能会有节点一直在Candidate中，导致死循环。

代码

    def MiniSpanTree_prim(self,vName):  #最小生成树-prim
        i=self.GetIndex(vName) 
        if i==-1:
            return None
        for i in range(self.vertexCount)