POJ-1258【Prim算法模板】

最新推荐文章于 2021-07-28 09:52:27 发布

原创最新推荐文章于 2021-07-28 09:52:27 发布 · 257 阅读

0 ·

CC 4.0 BY-SA版权

图论专栏收录该内容

8 篇文章

订阅专栏

本文介绍了一种使用Prim算法寻找最小生成树的方法，并提供了一个完整的C++代码示例。通过该算法可以有效地解决网络中边权值构成的图的最小生成树问题。

//
4
0 4 9 21
4 0 8 17
9 8 0 16
21 17 16 0
Output
28

学习Prim，类似迪杰斯特拉。。

#include<iostream>  
#include<cstdio>  
#include<cstring>  
#define inf 0x3f3f3f3f  
using namespace std;  
int Map[600][600],dis[600],vis[600],z,n,m;  
void Prim()  
{  
    int ma;  
    int pos;  
    memset(vis,0,sizeof(vis));  
    for(int i=1;i<=n;i++)  
        dis[i]=Map[1][i];  
    vis[1]=1;  
    for(int i=1;i<n;i++)//  
    {  
        ma=inf;  
        for(int j=1;j<=n;j++)  
        {  
            if(ma>dis[j]&&!vis[j])  
            {  
                ma=dis[j];  
                pos=j;  
            }  
        }  
        z+=ma;  
        vis[pos]=1;  
        for(int j=1;j<=n;j++)  
        {  
            if(dis[j]>Map[pos][j]&&!vis[j])  
            {  
                dis[j]=Map[pos][j];  
            }  
        }  
    }  
     cout<<z<<endl;  
}  
int main()  
{  
    int i,j,k;  
    while(cin>>n&&n)  
    {  
        z=0;  
        for(i=1;i<=n;i++)  
        {  
            for(j=1;j<=n;j++)  
            {  
                cin>>Map[i][j];  
            }  
        }  
        Prim();  
    }  
    return 0;  
}