Prim模板

C++ 代码

#include<bits/stdc++.h>
using namespace std;
const int N=510,INF=0x3f3f3f3f; 
int n,m,g[N][N],dist[N];
bool st[N];
int Prim()
{
    memset(dist,0x3f,sizeof dist);
    int res=0;
    for(int i=1;i<=n;i++)
    {
        int t=-1;
        for(int j=1;j<=n;j++)
            if(!st[j] and (t==-1 or dist[t]>dist[j]))
                t=j;
        if(i!=1 and dist[t]==INF) return INF;
        if(i!=1) 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()
{
    cin>>n;
    for(int i=1;i<=n;i++)
        for(int j=1;j<=n;j++)
            cin>>g[i][j];
    int t=Prim();
    if(t==INF) cout<<"impossible";
    else cout<<t;
    return 0;
}