模仿高手的代码

import sys

N = 101
inf = 0x3f3f3f3f


def prim(n, m, mapp):
    vis = [False] * N
    dis = [inf] * N
    sum_val = 0
    for i in range(1, n + 1):
        dis[i] = mapp[1][i]
    vis[1] = True
    dis[1] = 0
    for i in range(1, n):
        minn = inf
        u = 0
        for j in range(1, n + 1):
            if not vis[j] and dis[j] < minn:
                minn = dis[j]
                u = j
        vis[u] = True
        sum_val += minn
        for j in range(1, n + 1):
            if not vis[j] and mapp[u][j] < dis[j]:
                dis[j] = mapp[u][j]
    return sum_val


if __name__ == '__main__':
    n, m = map(int, input().split())
    mapp = [[inf] * N for _ in range(N)]
    ss = 0
    for _ in range(m):
        a, b, c = map(int, input().split())
        mapp[a][b] = mapp[b][a] = c
        ss += c
    result = prim(n, m, mapp)
    print(ss - result)