Prim的朴素做法

申明

再次申明这个专栏只是分享博主学习图论时的顺序，给新手提供一个入门的好方法（博主现在也是新手所以博主写的代码适合C转C++的新手学习），具体算法逻辑需要自己学习，代码区域有部分解释,因此有错误可以私信或者直接在评论区骂。

前言

Prim算法主要解决最小生成树问题，时间复杂度为On方，适用于稠密图，写法和Dijkstar的朴素算法差不多

题目链接

P3366 【模板】最小生成树 - 洛谷 | 计算机科学教育新生态 (luogu.com.cn)

最小生成树问题一定要注意连不连通

代码

#include <iostream>
#include <algorithm>
#include <cstdio>
#include <vector>
#include <cstring>
#include <queue>
using namespace std;
typedef pair<int,int>PII;
const int N=5001;
int g[N][N],dist[N];
bool st[N];
int n,m,num;
int Prim()
{
    int res=0;
    for(int i=1;i<=n;i++) dist[i]=0x3f3f3f3f;
    dist[1]=0;
    for(int i=0;i<n;i++)
    {
        int t=-1;
        for(int j=1;j<=n;j++)
            if(!st[j]&&(t==-1||dist[j]<dist[t])) 
                t=j;

        num++;
        if(dist[t]==0x3f3f3f3f) break;
        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);
    for(int i=1;i<=n;i++)
     for(int j=1;j<=n;j++)
      g[i][j]=0x3f3f3f3f;
    for(int i=1;i<=m;i++)
    {
        int u,v,w;
        scanf("%d %d %d",&u,&v,&w);
        g[u][v]=min(g[u][v],w);
        g[v][u]=min(g[v][u],w);
    }
    int res=Prim();
    if(num==n) printf("%d\n",res);
    else printf("orz");
    return 0;
}