首页 > 代码库 > POJ 1258 Agri-Net (prim最小生成树)

POJ 1258 Agri-Net (prim最小生成树)

最小生成树模板题

#include<bits/stdc++.h>
using namespace std;
int n,a;
int dist[120],m[120][120];
void prim()
{
    bool p[1020];
    for(int i=2;i<=n;i++)
    {
        p[i]=false;
        dist[i]=m[1][i];
    }
    dist[1]=0,p[1]=true;
    for(int i=1;i<=n-1;i++)
    {
        int min=INT_MAX,k=0;
        for(int j=1;j<=n;j++)
        {
            if(!p[j]&&dist[j]!=0&&dist[j]<min)
            {
                min=dist[j];
                k=j;
            }
        }
        if(k==0)
            return;
        p[k]=true;
        for(int j=1;j<=n;j++)
        {
            if(!p[j]&&m[k][j]!=0&&(dist[j]==0||dist[j]>m[k][j]))
                dist[j]=m[k][j];
        }
    }
}
int main()
{
    while(scanf("%d",&n)==1)
    {
        for(int i=1;i<=n;i++)
            for(int j=1;j<=n;j++)
            {
                scanf("%d",&a);
                m[i][j]=a;
            }
        prim();
        int sum=0;
        for(int i=1;i<=n;i++)
            sum+=dist[i];
        printf("%d\n",sum);
    }
    return 0;
}