最小生成树

#include <iostream>
#include <cstdio>
#include <cstring>
#define MAX 0x3f3f3f3f

using namespace std;

int map[1123][1123];
int vis[1123];
int lowcost[1123];
int flag, sum;

void prim_MST(int n)
{
    flag = 0;
    sum = 0;

    int k = 1;
    for(int i = 1; i <= n; i++)
    {
        lowcost[i] = map[i][1]; /// 初始化
    }
    vis[1] = 1;
    for(int i = 2; i <= n; i++)
    {
        int t = MAX;
        for(int j = 1; j <= n; j++)
        {
            if(!vis[j] && t > lowcost[j]) /// 找到点未访问且权值最小的边
            {
                t = lowcost[j];
                k = j; /// 记录该点
            }
        }
        if(t == MAX) /// 没有符合条件的点即此图不连通
        {
            //cout << "-1" << endl;
            flag = 1;
            return ;
        }

        vis[k] = 1;
        sum+=t; /// 权值累加

        for(int j = 1; j <= n; j++)
        {
            if(!vis[j] && map[j][k] < lowcost[j]) /// 如果此时点未访问且权值小于此时lowcost的值
            {
                lowcost[j] = map[j][k]; /// 更新
            }
        }
    }
}

int main()
{
    int n, m, cost;

    while(cin >> n >> m)
    {

        ///map初始化为无穷大，不是初始化为0；
        for(int i = 1;i <= n;i++)
        {
            for(int j = 1;j <= n;j++)
            {
                if(i == j)
                    map[i][j] = 0;
                else
                    map[i][j] = MAX;
            }
        }
        memset(vis, 0, sizeof(vis));

        while(m--)
        {
            int u, v;
            cin >> u >> v >> cost;
            if(map[u][v] > cost) /// 窝草 重边是什么鬼！！！坑死了！！！！QAQ
            {
                map[u][v] = map[v][u] = cost; 
            }
        }
        prim_MST(n);

        if(!flag)
        {
            cout << sum << endl;
        }
    }
    return 0;
}