858. Prim算法求最小生成树

朴素版

#include <bits/stdc++.h>
using namespace std;

const int N = 510, M = 100010 , INF = 0x3f3f3f3f;
int n,m;
//dist表示当前连通块内和连通块外连接到点的每一个边权的长度
int g[N][N],dist[N];//g存图（稠密图用 枚举来存）
bool st[N];//表示每一个点是否在连通块内

int prim()
{
    //先把距离初始化为正无穷
    memset(dist,0x3f,sizeof dist);
    dist[1] = 0; //先选第一个点，初始化为0
    int res = 0;//定义一下答案(边权和）
    for(int i = 0 ; i < n ; i++)//一共要加n个点
    {
        int t = -1;//t表示当前边权最小的一个点(初始化成-1)
        for(int j = 1 ; j <= n ; j++)//枚举一下其他所有点
        {
            //如果这个点没有在集合中，或者这个点还没有被选过，或者当前点的长度小于原来的
            if(!st[j] && (t == -1 || dist[t] > dist[j]))
            {
                t = j;
            }
        }
        //如果当前最小的点都是无穷，那就是不连通，那就直接返回
        if(dist[t] == INF) return INF;

        st[t] = true;
        res = res + dist[t];
        for(int j = 1 ; j <= n ; j++)
        {
            //更新点j到当前连通块的距离
            // 如果通过点t到达点j的距离小于当前已知的点j到连通块的距离
            // 那么就更新点j到连通块的距离为通过点t到达的距离。
            dist[j] = min(dist[j],g[t][j]);
        }
    }
    return res;
}

signed main()
{
    cin>>n>>m;
    memset(g,0x3f,sizeof g);//首先先把所有的边初始化为正无穷
    while(m--)//读入所有边
    {
        int a,b,c;
        cin>>a>>b>>c;
        //两条边之间有多条连线。只保存最小的就行
        g[a][b] = g[b][a] = min(g[a][b],c);
    }
    int t = prim();
    if(t == INF) cout<<"impossible"<<endl;
    else cout<<t<<endl;
    return 0;
}

#include <cstring>
#include <iostream>
#include <algorithm>

using namespace std;

const int N = 510, INF = 0x3f3f3f3f;

int n, m;
int g[N][N];
int dist[N];
bool st[N];


int prim()
{
    memset(dist, 0x3f, sizeof dist);

    int res = 0;
    for (int i = 0; i < n; i ++ )
    {
        int t = -1;
        for (int j = 1; j <= n; j ++ )
            if (!st[j] && (t == -1 || dist[t] > dist[j]))
                t = j;

        if (i && dist[t] == INF) return INF;

        if (i) res += dist[t];
        st[t] = true;

        for (int j = 1; j <= n; j ++ ) dist[j] = min(dist[j], g[t][j]);
    }

    return res;
}


int main()
{
    scanf("%d%d", &n, &m);

    memset(g, 0x3f, sizeof g);

    while (m -- )
    {
        int a, b, c;
        scanf("%d%d%d", &a, &b, &c);
        g[a][b] = g[b][a] = min(g[a][b], c);
    }

    int t = prim();

    if (t == INF) puts("impossible");
    else printf("%d\n", t);

    return 0;
}