最小生成树Prim算法贪心法python实现

最小生成树Prim算法贪心法python实现

#!/usr/bin/env python
# -*- coding: utf-8 -*-


# prim算法
def update(a,b, nodepair, G, n):
    # 函数实现代码开始
    # 请在其中写入你的代码
    if b == 0:
        return nodepair
    nodepair[b][0] = a
    nodepair[b][1] = G[a][b]
    # 函数实现代码结束
    return nodepair


def prim(G, n):
    INF = 100000
    visit = [-1] * n  # 遍历情况记录数组
    nodepair = [[-1, INF] for i in range(n)]  # 节点伴随二元数组 第一个表示前一个索引preIndex，第二个表示距离
    # 先取第一个节点
    visit[0] = 1
    print(0, nodepair[0])  # 打印第一个节点
    # 更新节点伴随二元数组
    nodepair = update(0,0, nodepair, G, n)
    # print(nodepair)
    # 函数实现代码开始
    # 请在其中写入你的代码
    #对visit进行遍历

    for key in range(1,n):
        tmp = INF
        for key1 in range(n):
            for key2 in range(n):
                if visit[key1]==1 and visit[key2]==-1:
                    if G[key1][key2] != 0:
                        if G[key1][key2] < tmp:
                            a = key1
                            b = key2
                            tmp = G[key1][key2]
        visit[b] = 1
        nodepair = update(a,b, nodepair, G, n)
        print(b, nodepair[b])

    # 函数实现代码结束

    return nodepair


if __name__ == '__main__':
    Glist = input().split('-')
    G = []
    for item in Glist:
        row = item.split(',')
        g = [int(i) for i in row]
        G.append(g)
    n = len(G)
    prim(G, n)